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