support sovits v2Pro v2ProPlus

2 months ago · e920b31840
parent 921ac6c41a
commit e920b31840
1 changed files with 1 additions and 1 deletions
--- a/GPT_SoVITS/AR/models/t2s_lightning_module.py
+++ b/GPT_SoVITS/AR/models/t2s_lightning_module.py
@ -28,7 +28,7 @@ class Text2SemanticLightningModule(LightningModule):
                self.load_state_dict(
                    torch.load(
                        pretrained_s1,
-                        map_location="cpu",
+                        map_location="cpu", weights_only=False,
                    )["weight"],
                )
            )