Update config.py
Browse files
config.py
CHANGED
@@ -8,43 +8,6 @@ from tools.i18n.i18n import I18nAuto
|
|
8 |
|
9 |
i18n = I18nAuto(language=os.environ.get("language", "Auto"))
|
10 |
|
11 |
-
from huggingface_hub import hf_hub_download
|
12 |
-
|
13 |
-
# 1. 定义仓库信息和本地目标路径
|
14 |
-
# ----------------------------------------------------
|
15 |
-
# 您的远程模型仓库
|
16 |
-
repo_id = "l73jiang/Seia-GPT-SOVITS-ProPlus" # <-- 请替换成您的用户名和仓库名
|
17 |
-
# 您希望文件被存放在 Space 中的哪个文件夹
|
18 |
-
target_dir = "pretrained_models"
|
19 |
-
|
20 |
-
# 2. 将所有需要下载的模型文件名放入一个列表
|
21 |
-
# ----------------------------------------------------
|
22 |
-
files_to_download = [
|
23 |
-
"Seia-e15.ckpt",
|
24 |
-
"Seia_e8_s240.pth" # <-- 新增了第二个模型文件
|
25 |
-
]
|
26 |
-
|
27 |
-
# 3. 确保目标文件夹存在(这个操作只需执行一次)
|
28 |
-
# ----------------------------------------------------
|
29 |
-
os.makedirs(target_dir, exist_ok=True)
|
30 |
-
print(f"目标文件夹 '{target_dir}' 已准备就绪。")
|
31 |
-
|
32 |
-
# 4. 循环遍历列表,下载每一个文件
|
33 |
-
# ----------------------------------------------------
|
34 |
-
for filename in files_to_download:
|
35 |
-
print(f"-> 开始从仓库 '{repo_id}' 下载 '{filename}'...")
|
36 |
-
try:
|
37 |
-
hf_hub_download(
|
38 |
-
repo_id=repo_id,
|
39 |
-
filename=filename,
|
40 |
-
local_dir=target_dir, # 所有文件都下载到同一个目标文件夹
|
41 |
-
local_dir_use_symlinks=False
|
42 |
-
)
|
43 |
-
print(f" 文件 '{filename}' 下载成功!")
|
44 |
-
except Exception as e:
|
45 |
-
# 增加一个错误处理,这样如果某个文件下载失败,应用不会直接崩溃
|
46 |
-
print(f" !!! 下载文件 '{filename}' 时发生错误: {e}")
|
47 |
-
|
48 |
|
49 |
pretrained_sovits_name = {
|
50 |
"v1": "pretrained_models/s2G488k.pth",
|
|
|
8 |
|
9 |
i18n = I18nAuto(language=os.environ.get("language", "Auto"))
|
10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
pretrained_sovits_name = {
|
13 |
"v1": "pretrained_models/s2G488k.pth",
|