From 32e88bc3c5594015e741e84887a48ced238f4198 Mon Sep 17 00:00:00 2001
From: Daniel Strobusch <1847260+dastrobu@users.noreply.github.com>
Date: Tue, 2 Jan 2024 17:18:12 +0100
Subject: [PATCH] make parameter naming consistent with other examples. (#214)

---
 llms/mistral/mistral.py | 2 +-
 llms/mixtral/mixtral.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/llms/mistral/mistral.py b/llms/mistral/mistral.py
index 105a7988..9b9a602a 100644
--- 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,
diff --git a/llms/mixtral/mixtral.py b/llms/mixtral/mixtral.py
index 6aee4031..b1f14706 100644
--- 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,