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",
type=str,
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.
""",
)