blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-05 22:20:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

modified: GPT_SoVITS/TTS_infer_pack/TTS.py

Browse Source
This commit is contained in:
ChasonJiang 2025-04-21 23:08:15 +08:00
parent b5f2709675
commit 4bd2625e7c
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
GPT_SoVITS/TTS_infer_pack/TTS.py
View File

@ -465,10 +465,10 @@ class TTS:
def init_vits_weights(self, weights_path: str):
self.configs.vits_weights_path = weights_path
version, model_version, if_lora_v3 = get_sovits_version_from_path_fast(weights_path)
path_sovits_v3 = self.configs.default_configs["v3"]["vits_weights_path"]
path_sovits = self.configs.default_configs[model_version]["vits_weights_path"]
if if_lora_v3 == True and os.path.exists(path_sovits_v3) == False:
info = path_sovits_v3 + i18n("SoVITS V3 底模缺失，无法加载相应 LoRA 权重")
if if_lora_v3 == True and os.path.exists(path_sovits) == False:
info = path_sovits + i18n("SoVITS %s 底模缺失，无法加载相应 LoRA 权重"%model_version)
raise FileExistsError(info)
# dict_s2 = torch.load(weights_path, map_location=self.configs.device,weights_only=False)
@ -523,7 +523,6 @@ class TTS:
f"Loading VITS weights from {weights_path}. {vits_model.load_state_dict(dict_s2['weight'], strict=False)}"
)
else:
path_sovits = self.configs.default_configs[model_version]["vits_weights_path"]
print(
f"Loading VITS pretrained weights from {weights_path}. {vits_model.load_state_dict(load_sovits_new(path_sovits)['weight'], strict=False)}"
)