blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-12-06 00:56:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Modelscope (#2679)

Browse Source
This commit is contained in:
XXXXRT666 2025-12-01 03:13:15 +00:00 committed by GitHub
parent 92d2d337fd
commit 857799276c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tools/asr/fasterwhisper_asr.py
View File

@ -52,7 +52,7 @@ def download_model(model_size: str):
if "distil" in model_size: if "distil" in model_size:
if "3.5" in model_size: if "3.5" in model_size:
repo_id = "distil-whisper/distil-large-v3.5-ct2" repo_id = "distil-whisper/distil-large-v3.5-ct2"
model_path = "tools/asr/models/faster-whisper-distil-large-v3.5" model_path = "tools/asr/models/faster-distil-whisper-large-v3.5"
else: else:
repo_id = "Systran/faster-{}-whisper-{}".format(*model_size.split("-", maxsplit=1)) repo_id = "Systran/faster-{}-whisper-{}".format(*model_size.split("-", maxsplit=1))
elif model_size == "large-v3-turbo": elif model_size == "large-v3-turbo":
@ -61,14 +61,11 @@ def download_model(model_size: str):
else: else:
repo_id = f"Systran/faster-whisper-{model_size}" repo_id = f"Systran/faster-whisper-{model_size}"
model_path = ( model_path = (
model_path model_path or f"tools/asr/models/{repo_id.replace('Systran/', '').replace('distil-whisper/', '', 1)}"
or f"tools/asr/models/{repo_id.replace('Systran/', '').replace('distil-whisper/', '', 1)}".replace(
"distil-whisper", "whisper-distil"
)
) )
else: else:
repo_id = "XXXXRT/faster-whisper" repo_id = "XXXXRT/faster-whisper"
model_path = f"tools/asr/models/faster-whisper-{model_size}".replace("distil-whisper", "whisper-distil") model_path = "tools/asr/models"
files: list[str] = [ files: list[str] = [
"config.json", "config.json",
@ -76,7 +73,7 @@ def download_model(model_size: str):
"tokenizer.json", "tokenizer.json",
"vocabulary.txt", "vocabulary.txt",
] ]
if model_size == "large-v3" or "distil" in model_size: if "large-v3" in model_size or "distil" in model_size:
files.append("preprocessor_config.json") files.append("preprocessor_config.json")
files.append("vocabulary.json") files.append("vocabulary.json")