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

2025-09-01 04:14:38 +08:00 · 2024-01-02 17:18:12 +01:00
parent e632d7aaaa
commit 85258b2be7
2 changed files with 2 additions and 2 deletions
--- 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,