Skip to content

Commit f9d5c03

Browse files
Set weights_only=False in export_serialize.deserialize_torch_artifact
1 parent c734ad4 commit f9d5c03

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

exir/serde/export_serialize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -327,7 +327,7 @@ def deserialize_torch_artifact(
327327
return {}
328328
buffer = io.BytesIO(serialized)
329329
buffer.seek(0)
330-
artifact = torch.load(buffer)
330+
artifact = torch.load(buffer, weights_only=False)
331331
assert isinstance(artifact, (tuple, dict))
332332
return artifact
333333

0 commit comments

Comments
 (0)