mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-08-08 00:06:21 +08:00
28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
# Download moda ASR related models
|
|
from modelscope import snapshot_download
|
|
import shutil
|
|
import os
|
|
import time
|
|
from modelscope.hub.snapshot_download import snapshot_download
|
|
|
|
def safe_download(model_id, revision='v2.0.4', max_retries=3):
|
|
cache_base = os.path.expanduser(f'~/.cache/modelscope/hub/{model_id.replace("/", "/")}')
|
|
if os.path.exists(cache_base):
|
|
shutil.rmtree(cache_base) # Remove possibly corrupted model cache
|
|
|
|
for attempt in range(max_retries):
|
|
try:
|
|
print(f"Attempting download for: {model_id} (Attempt {attempt + 1})")
|
|
model_dir = snapshot_download(model_id, revision=revision)
|
|
print(f"✅ Downloaded {model_id} to {model_dir}")
|
|
return model_dir
|
|
except Exception as e:
|
|
print(f"❌ Failed on attempt {attempt + 1} for {model_id}: {e}")
|
|
time.sleep(5)
|
|
raise RuntimeError(f"❌ All attempts failed for {model_id}")
|
|
|
|
# Model downloads
|
|
safe_download('damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch')
|
|
safe_download('damo/speech_fsmn_vad_zh-cn-16k-common-pytorch')
|
|
safe_download('damo/punc_ct-transformer_zh-cn-common-vocab272727-pytorch')
|