mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-09-01 04:14:38 +08:00
make parameter naming consistent with other examples.
This commit is contained in:
@@ -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,
|
||||
|
@@ -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,
|
||||
|
Reference in New Issue
Block a user