add base option

This commit is contained in:
root 2024-04-30 03:05:14 +00:00
parent c292dc6e2a
commit 844ca42a39

View File

@ -215,7 +215,7 @@ def get_parser():
"--model-name", "--model-name",
type=str, type=str,
default="large-v2", default="large-v2",
choices=["large-v2", "large-v3", "medium", "small", "tiny"], choices=["large-v2", "large-v3", "medium", "base", "small", "tiny"],
help="""The model name to use. help="""The model name to use.
""", """,
) )