25 lines
876 B
Python
25 lines
876 B
Python
#!/usr/bin/env python3
|
|
"""Download ECAPA-TDNN model via ModelScope to /share/models/ecapa-tdnn/"""
|
|
import os, sys, time
|
|
|
|
MODEL_ID = 'iic/speech_ecapa-tdnn_sv_en_voxceleb_16k'
|
|
TARGET = '/share/models/ecapa-tdnn'
|
|
|
|
os.makedirs(TARGET, exist_ok=True)
|
|
|
|
# Try ModelScope first
|
|
for attempt in range(3):
|
|
try:
|
|
from modelscope import snapshot_download
|
|
print(f"Attempt {attempt+1}: ModelScope download...")
|
|
snapshot_download(MODEL_ID, local_dir=TARGET)
|
|
print("SUCCESS via ModelScope")
|
|
sys.exit(0)
|
|
except Exception as e:
|
|
print(f"ModelScope failed: {e}")
|
|
time.sleep(10)
|
|
|
|
# Fallback: try HF mirror
|
|
print("Trying HF mirror...")
|
|
os.system(f'export HF_ENDPOINT=https://hf-mirror.com && python3 -c "from huggingface_hub import snapshot_download; snapshot_download(\'speechbrain/spkrec-ecapa-voxceleb\', local_dir=\'{TARGET}\')"')
|