fix

2026-07-23 11:51:07 +09:00
parent 4f3bcfd252
commit 9cfaa27f3f
5 changed files with 87 additions and 17 deletions
--- a/style_bert_vits2/models/utils/checkpoints.py
+++ b/style_bert_vits2/models/utils/checkpoints.py
@@ -32,7 +32,9 @@ def load_checkpoint(
    """

    assert os.path.isfile(checkpoint_path)
-    checkpoint_dict = torch.load(checkpoint_path, map_location=device)
+    checkpoint_dict = torch.load(
+        checkpoint_path, map_location=device, weights_only=True
+    )
    iteration = checkpoint_dict["iteration"]
    learning_rate = checkpoint_dict["learning_rate"]
    logger.info(