from huggingface_hub import snapshot_download
import os

# [关键修改] 将目标文件夹指向程序期望的正确路径
target = "GPT_SoVITS/pretrained_models"
os.makedirs(target, exist_ok=True)

try:
    snapshot_download(
        repo_id="lj1995/GPT-SoVITS",
        repo_type="model",
        local_dir=target,
        # 注意：这里我们下载全部预训练模型，而不仅仅是sv和chinese
        # allow_patterns=["sv/*", "chinese*"],  # 暂时注释掉，确保所有基础模型都被下载
    )
    print("Support models downloaded to ./GPT_SoVITS/pretrained_models")
except Exception as e:
    print("Skipping support model download:", e)