update torch load method

microsoft · Jan 30, 2025 · fa1c09b · fa1c09b
1 parent e3f4a6e
commit fa1c09b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/anomaly_detector/multivariate/model.py b/src/anomaly_detector/multivariate/model.py
@@ -151,7 +151,7 @@ def save_checkpoint(self):
         self.model.to(self.config.device)
 
     def load_checkpoint(self, model_path):
-        ckpt = torch.load(model_path)
+        ckpt = torch.load(model_path, weights_only=True)
         self.config = ckpt["config"]
         self.model = MultivariateGraphAttnDetector(self.config)
         self.model.load_state_dict(ckpt["state_dict"])