fix torch.load

4 months ago · 13573a1b06
parent fef65d40fe
commit 13573a1b06
1 changed files with 1 additions and 1 deletions
--- a/GPT_SoVITS/prepare_datasets/3-get-semantic.py
+++ b/GPT_SoVITS/prepare_datasets/3-get-semantic.py
@ -81,7 +81,7 @@ if os.path.exists(semantic_path) == False:
    # utils.load_checkpoint(pretrained_s2G, vq_model, None, True)
    print(
        vq_model.load_state_dict(
-            torch.load(pretrained_s2G, map_location="cpu")["weight"], strict=False
+            torch.load(pretrained_s2G, map_location="cpu", weights_only=False)["weight"], strict=False
        )
    )