Refactor: Format Code with Ruff and Update Deprecated G2PW Link (#2255)

* ruff check --fix

* ruff format --line-length 120 --target-version py39

* Change the link for G2PW Model

* update pytorch version and colab
This commit is contained in:
XXXXRT666
2025-04-07 09:42:47 +01:00
committed by GitHub
parent 9da7e17efe
commit 53cac93589
132 changed files with 8185 additions and 6648 deletions

View File

@@ -1,33 +1,36 @@
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"]
"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'
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']
},
"达摩 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']
"lang": ["auto", "zh", "en", "ja", "ko", "yue"],
"size": check_fw_local_models(),
"path": "fasterwhisper_asr.py",
"precision": ["float32", "float16", "int8"],
},
}