fix(config): Fix errors when running inference webui directly (#2426)

This commit is contained in:
SapphireLab 2025-06-05 00:26:44 +08:00 committed by GitHub
parent 09e9961a0d
commit 2ff2cf5ba1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -4,7 +4,7 @@ import os
import torch,re
from tools.i18n.i18n import I18nAuto, scan_language_list
i18n = I18nAuto(language=os.environ["language"])
i18n = I18nAuto(language=os.environ.get("language", "Auto"))
pretrained_sovits_name = {
@ -67,6 +67,7 @@ def get_weights_names():
for key in name2sovits_path:
if os.path.exists(name2sovits_path[key]):SoVITS_names.append(key)
for path in SoVITS_weight_root:
if not os.path.exists(path):continue
for name in os.listdir(path):
if name.endswith(".pth"):
SoVITS_names.append("%s/%s" % (path, name))
@ -74,6 +75,7 @@ def get_weights_names():
for key in name2gpt_path:
if os.path.exists(name2gpt_path[key]):GPT_names.append(key)
for path in GPT_weight_root:
if not os.path.exists(path):continue
for name in os.listdir(path):
if name.endswith(".ckpt"):
GPT_names.append("%s/%s" % (path, name))