From f9d5c03cf9507436f6d6c5f6e3d752ba546a38c2 Mon Sep 17 00:00:00 2001
From: Mikayla Gawarecki <mikaylagawarecki@gmail.com>
Date: Fri, 15 Nov 2024 14:53:10 -0800
Subject: [PATCH] Set weights_only=False in
 export_serialize.deserialize_torch_artifact

---
 exir/serde/export_serialize.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exir/serde/export_serialize.py b/exir/serde/export_serialize.py
index 08cd03adcea..6327cd5da19 100644
--- a/exir/serde/export_serialize.py
+++ b/exir/serde/export_serialize.py
@@ -327,7 +327,7 @@ def deserialize_torch_artifact(
         return {}
     buffer = io.BytesIO(serialized)
     buffer.seek(0)
-    artifact = torch.load(buffer)
+    artifact = torch.load(buffer, weights_only=False)
     assert isinstance(artifact, (tuple, dict))
     return artifact