删除加载v3sovits模型缺少enc_q告警

4 months ago · 7abae557fb
parent 6a60e5edb1
commit 7abae557fb
1 changed files with 2 additions and 3 deletions
--- a/GPT_SoVITS/TTS_infer_pack/TTS.py
+++ b/GPT_SoVITS/TTS_infer_pack/TTS.py
@ -462,8 +462,6 @@ class TTS:
                n_speakers=self.configs.n_speakers,
                **kwargs
            )
-            if hasattr(vits_model, "enc_q"):
-                del vits_model.enc_q
            self.configs.is_v3_synthesizer = False
        else:
            vits_model = SynthesizerTrnV3(
@ -474,7 +472,8 @@ class TTS:
            )
            self.configs.is_v3_synthesizer = True
            self.init_bigvgan()
-            
+            if "pretrained" not in weights_path and hasattr(vits_model, "enc_q"):
+                del vits_model.enc_q

        if if_lora_v3==False:
            print(f"Loading VITS weights from {weights_path}. {vits_model.load_state_dict(dict_s2['weight'], strict=False)}")