dangthr commited on
Commit
a1fb0be
·
verified ·
1 Parent(s): 160e694

Create downloader.py

Browse files
Files changed (1) hide show
  1. downloader.py +20 -0
downloader.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from huggingface_hub import snapshot_download
2
+
3
+ # 定义 Hugging Face 仓库 ID
4
+ REPO_ID = "FrancisRing/StableAvatar"
5
+
6
+ # 下载模型和相关文件到默认缓存目录
7
+ print(f"开始下载模型从 {REPO_ID} 到 Hugging Face 默认缓存目录...")
8
+ repo_root = snapshot_download(
9
+ repo_id=REPO_ID,
10
+ allow_patterns=[
11
+ "StableAvatar-1.3B/*", # Transformer 模型文件
12
+ "Wan2.1-Fun-V1.1-1.3B-InP/*", # Wan 模型相关文件
13
+ "wav2vec2-base-960h/*", # Wav2Vec2 模型
14
+ "assets/**", # 其他资源文件
15
+ "Kim_Vocal_2.onnx", # 人声分离模型
16
+ ],
17
+ local_dir_use_symlinks=False, # 确保文件实际下载而不是创建符号链接
18
+ )
19
+
20
+ print(f"模型已下载到: {repo_root}")