make parameter naming consistent with other examples. (#214)

2024-01-02 17:18:12 +01:00 · 2024-01-02 17:18:12 +01:00 · 85258b2be7
commit 85258b2be7
parent e632d7aaaa
2 changed files with 2 additions and 2 deletions
--- a/llms/mistral/mistral.py
+++ b/llms/mistral/mistral.py
@ -254,7 +254,7 @@ if __name__ == "__main__":
        default=0.0,
    )
    parser.add_argument(
-        "--tokens_per_eval",
+        "--tokens-per-eval",
        help="The batch size of tokens to generate.",
        type=int,
        default=10,
--- a/llms/mixtral/mixtral.py
+++ b/llms/mixtral/mixtral.py
@ -299,7 +299,7 @@ if __name__ == "__main__":
        default="In the beginning the Universe was created.",
    )
    parser.add_argument(
-        "--max_tokens",
+        "--max-tokens",
        "-m",
        type=int,
        default=100,