Update RMVPE.py

Bebra777228 · Aug 4, 2024 · 8256fe2 · 8256fe2
1 parent 56bf99a
commit 8256fe2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/infer_pack/predictor/RMVPE.py b/src/infer_pack/predictor/RMVPE.py
@@ -278,7 +278,7 @@ def __init__(self, model_path, is_half, device=None):
         self.device = device
         self.mel_extractor = MelSpectrogram(is_half, 128, 16000, 1024, 160, None, 30, 8000).to(device)
         model = E2E(4, 1, (2, 2))
-        ckpt = torch.load(model_path, map_location="cpu")
+        ckpt = torch.load(model_path, map_location="cpu", weights_only=True)
         model.load_state_dict(ckpt)
         model.eval()
         if is_half: