mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-27 18:54:18 +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", "--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", "small", "base", "tiny"],
help="""The model name to use. help="""The model name to use.
""", """,
) )

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

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

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

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

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

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

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