mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-08-18 15:59:51 +08:00
45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
import os
|
|
|
|
|
|
def check_fw_local_models():
|
|
"""
|
|
启动时检查本地是否有 Faster Whisper 模型.
|
|
"""
|
|
model_size_list = [
|
|
"medium",
|
|
"medium.en",
|
|
"distil-large-v2",
|
|
"distil-large-v3",
|
|
"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
|
|
|
|
|
|
def get_models():
|
|
model_size_list = [
|
|
"medium",
|
|
"medium.en",
|
|
"distil-large-v2",
|
|
"distil-large-v3",
|
|
"large-v1",
|
|
"large-v2",
|
|
"large-v3",
|
|
]
|
|
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": get_models(),
|
|
"path": "fasterwhisper_asr.py",
|
|
"precision": ["float32", "float16", "int8"],
|
|
},
|
|
}
|