mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-24 17:31:18 +08:00
fix (#1192)
This commit is contained in:
parent
f2619f507c
commit
9183fe8b6d
@ -45,7 +45,7 @@ def setup_arg_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--extra-eos-token",
|
"--extra-eos-token",
|
||||||
type=str,
|
type=str,
|
||||||
default=None,
|
default=(),
|
||||||
nargs="+",
|
nargs="+",
|
||||||
help="Add tokens in the list of eos tokens that stop generation.",
|
help="Add tokens in the list of eos tokens that stop generation.",
|
||||||
)
|
)
|
||||||
|
Loading…
Reference in New Issue
Block a user