Skip to content

Commit

Permalink
Set weights_only to False for xser.load (#8259)
Browse files Browse the repository at this point in the history
Co-authored-by: root <root@t1v-n-a83b02ef-w-0.us-central2-b.c.tpu-pytorch.internal>
  • Loading branch information
JackCaoG and root authored Oct 15, 2024
1 parent ecdc68b commit f71c02d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch_xla/utils/serialization.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit f71c02d

Please sign in to comment.