mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-30 02:53:41 +08:00
Run hf_t5 with any model
This commit is contained in:
parent
0779417903
commit
305a52dde8
@ -28,9 +28,6 @@ if __name__ == "__main__":
|
|||||||
)
|
)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--model",
|
"--model",
|
||||||
choices=[
|
|
||||||
"t5-small",
|
|
||||||
],
|
|
||||||
default="t5-small",
|
default="t5-small",
|
||||||
help="The huggingface name of the T5 model to save.",
|
help="The huggingface name of the T5 model to save.",
|
||||||
)
|
)
|
||||||
|
Loading…
Reference in New Issue
Block a user