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

Add base choice to model_name argument for whisper model. (#1573)

Browse Source 
Co-authored-by: dadoou <dadoou@yandex.com>
This commit is contained in:
Dadoou 2024-04-02 09:47:38 +08:00 committed by GitHub
parent 42de459110
commit 6cbddaa8e3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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.
""",
)