from huggingface_hub import snapshot_download

# 定义 Hugging Face 仓库 ID
REPO_ID = "FrancisRing/StableAvatar"

# 下载模型和相关文件到默认缓存目录
print(f"开始下载模型从 {REPO_ID} 到 Hugging Face 默认缓存目录...")
repo_root = snapshot_download(
    repo_id=REPO_ID,
    allow_patterns=[
        "StableAvatar-1.3B/*",          # Transformer 模型文件
        "Wan2.1-Fun-V1.1-1.3B-InP/*",   # Wan 模型相关文件
        "wav2vec2-base-960h/*",         # Wav2Vec2 模型
        "assets/**",                    # 其他资源文件
        "Kim_Vocal_2.onnx",            # 人声分离模型
    ],
    local_dir_use_symlinks=False,  # 确保文件实际下载而不是创建符号链接
)

print(f"模型已下载到: {repo_root}")