diff --git a/tools/asr/config.py b/tools/asr/config.py index 097aa643..fdff7518 100644 --- a/tools/asr/config.py +++ b/tools/asr/config.py @@ -15,7 +15,7 @@ def get_models(): asr_dict = { "达摩 ASR (中文)": {"lang": ["zh", "yue"], "size": ["large"], "path": "funasr_asr.py", "precision": ["float32"]}, "Faster Whisper (多语种)": { - "lang": ["auto", "en", "ja", "ko", "yue"], + "lang": ["auto", "en", "ja", "ko"], "size": get_models(), "path": "fasterwhisper_asr.py", "precision": ["float32", "float16", "int8"], diff --git a/tools/asr/fasterwhisper_asr.py b/tools/asr/fasterwhisper_asr.py index 1f98b840..72a4b82a 100644 --- a/tools/asr/fasterwhisper_asr.py +++ b/tools/asr/fasterwhisper_asr.py @@ -128,7 +128,7 @@ def execute_asr(input_folder, output_folder, model_path, language, precision): ) text = "" - if info.language == "zh": + if info.language in ["zh", "yue"]: print("检测为中文文本, 转 FunASR 处理") text = only_asr(file_path, language=info.language.lower())