zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 17:31:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Daniel Strobusch 2024-01-02 17:18:12 +01:00 committed by GitHub
parent e632d7aaaa
commit 85258b2be7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/mistral/mistral.py
View File

@ -254,7 +254,7 @@ if __name__ == "__main__":
default=0.0, default=0.0,
) )
parser.add_argument( parser.add_argument(
"--tokens_per_eval", "--tokens-per-eval",
help="The batch size of tokens to generate.", help="The batch size of tokens to generate.",
type=int, type=int,
default=10, default=10,

2
llms/mixtral/mixtral.py
View File

@ -299,7 +299,7 @@ if __name__ == "__main__":
default="In the beginning the Universe was created.", default="In the beginning the Universe was created.",
) )
parser.add_argument( parser.add_argument(
"--max_tokens", "--max-tokens",
"-m", "-m",
type=int, type=int,
default=100, default=100,