This commit is contained in:
XXXXRT666 2024-08-03 12:38:20 +08:00
parent 9de98a1d49
commit 60caceff5a

View File

@ -30,6 +30,8 @@ if os.path.exists("./gweight.txt"):
gweight_data = file.read() gweight_data = file.read()
gpt_path = os.environ.get( gpt_path = os.environ.get(
"gpt_path", gweight_data) "gpt_path", gweight_data)
if version == 'v2':
gpt_path = pretrained_gpt_name
else: else:
gpt_path = os.environ.get( gpt_path = os.environ.get(
"gpt_path", pretrained_gpt_name) "gpt_path", pretrained_gpt_name)
@ -38,6 +40,8 @@ if os.path.exists("./sweight.txt"):
with open("./sweight.txt", 'r', encoding="utf-8") as file: with open("./sweight.txt", 'r', encoding="utf-8") as file:
sweight_data = file.read() sweight_data = file.read()
sovits_path = os.environ.get("sovits_path", sweight_data) sovits_path = os.environ.get("sovits_path", sweight_data)
if version == 'v2':
sovits_path = pretrained_sovits_name
else: else:
sovits_path = os.environ.get("sovits_path", pretrained_sovits_name) sovits_path = os.environ.get("sovits_path", pretrained_sovits_name)
# gpt_path = os.environ.get( # gpt_path = os.environ.get(