Set weights_only to False for xser.load (#8259)

Co-authored-by: root <[email protected]>
pytorch · Oct 15, 2024 · f71c02d · f71c02d
1 parent ecdc68b
commit f71c02d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_xla/utils/serialization.py b/torch_xla/utils/serialization.py
@@ -84,7 +84,7 @@ def load(path):
   Returns:
     The loaded data.
   """
-  ref_data = torch.load(path)
+  ref_data = torch.load(path, weights_only=False)
   tensor_folder = _get_tensors_folder(path)
 
   def convert_fn(tensors):