mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-04-06 03:57:44 +08:00
37 lines
934 B
Python
37 lines
934 B
Python
import os
|
|
|
|
|
|
def check_fw_local_models():
|
|
"""
|
|
启动时检查本地是否有 Faster Whisper 模型.
|
|
"""
|
|
model_size_list = [
|
|
"tiny",
|
|
"tiny.en",
|
|
"base",
|
|
"base.en",
|
|
"small",
|
|
"small.en",
|
|
"medium",
|
|
"medium.en",
|
|
"large",
|
|
"large-v1",
|
|
"large-v2",
|
|
"large-v3",
|
|
]
|
|
for i, size in enumerate(model_size_list):
|
|
if os.path.exists(f"tools/asr/models/faster-whisper-{size}"):
|
|
model_size_list[i] = size + "-local"
|
|
return model_size_list
|
|
|
|
|
|
asr_dict = {
|
|
"达摩 ASR (中文)": {"lang": ["zh", "yue"], "size": ["large"], "path": "funasr_asr.py", "precision": ["float32"]},
|
|
"Faster Whisper (多语种)": {
|
|
"lang": ["auto", "zh", "en", "ja", "ko", "yue"],
|
|
"size": check_fw_local_models(),
|
|
"path": "fasterwhisper_asr.py",
|
|
"precision": ["float32", "float16", "int8"],
|
|
},
|
|
}
|