support sovits v2Pro v2ProPlus

2025-06-04 15:15:54 +08:00
parent 663c3cc6fc
commit b7c0c5ca87
4 changed files with 640 additions and 133 deletions
--- a/api.py
+++ b/api.py
@@ -374,7 +374,7 @@ hz = 50


 def get_gpt_weights(gpt_path):
-    dict_s1 = torch.load(gpt_path, map_location="cpu")
+    dict_s1 = torch.load(gpt_path, map_location="cpu", weights_only=False)
    config = dict_s1["config"]
    max_sec = config["data"]["max_sec"]
    t2s_model = Text2SemanticLightningModule(config, "****", is_train=False)