mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 10:16:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add base choice to model_name argument for whisper model.

Browse Source
This commit is contained in:
dadoou 2024-04-02 06:48:17 +08:00
parent 42de459110
commit 781ededa85
7 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/aishell/ASR/whisper/decode.py
View File

@ -214,7 +214,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", "small", "base", "tiny"],
help="""The model name to use.
""",
)

2
egs/aishell/ASR/whisper/train.py
View File

@ -147,7 +147,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", "small", "base", "tiny"],
help="""The model name to use.
""",
)

2
egs/multi_zh-hans/ASR/whisper/decode.py
View File

@ -214,7 +214,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", "small", "base", "tiny"],
help="""The model name to use.
""",
)

2
egs/multi_zh-hans/ASR/whisper/train.py
View File

@ -146,7 +146,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", "small", "base", "tiny"],
help="""The model name to use.
""",
)

2
egs/speechio/ASR/whisper/decode.py
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", "small", "base", "tiny"],
help="""The model name to use.
""",
)

2
egs/wenetspeech/ASR/whisper/decode.py
View File

@ -213,7 +213,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", "small", "base", "tiny"],
help="""The model name to use.
""",
)

2
egs/wenetspeech/ASR/whisper/train.py
View File

@ -145,7 +145,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", "small", "base", "tiny"],
help="""The model name to use.
""",
)