support wav2vec2-zh cli, test=asr

pull/2697/head
tianhao zhang 3 years ago
parent d9f28a87d4
commit 7fec275c7e

@ -14,3 +14,7 @@
import _locale import _locale
_locale._getdefaultlocale = (lambda *args: ['en_US', 'utf8']) _locale._getdefaultlocale = (lambda *args: ['en_US', 'utf8'])

@ -138,9 +138,9 @@ class SSLExecutor(BaseExecutor):
if model_type is None: if model_type is None:
if lang == 'en': if lang == 'en':
model_type = wav2vec2ASR_librispeech model_type = 'wav2vec2ASR_librispeech'
elif lang == 'zh': elif lang == 'zh':
model_type = wav2vec2ASR_aishell1 model_type = 'wav2vec2ASR_aishell1'
else: else:
logger.error( logger.error(
"invalid lang, please input --lang en or --lang zh") "invalid lang, please input --lang en or --lang zh")

Loading…
Cancel
Save