From 44f083eaf9188e36b79f731517f269594f030a35 Mon Sep 17 00:00:00 2001
From: root <root@t1v-n-a83b02ef-w-0.us-central2-b.c.tpu-pytorch.internal>
Date: Tue, 15 Oct 2024 01:13:41 +0000
Subject: [PATCH] Set weights_only to False for xser.load

---
 torch_xla/utils/serialization.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_xla/utils/serialization.py b/torch_xla/utils/serialization.py
index 7e2921d11f31..ed3797ad9456 100644
--- 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):