diff --git a/physicsnemo/models/module.py b/physicsnemo/models/module.py index 2b4f5aff44..5f7df56fad 100644 --- a/physicsnemo/models/module.py +++ b/physicsnemo/models/module.py @@ -541,7 +541,7 @@ def _save_process(module, args, metadata, mod_prefix="") -> None: with tempfile.NamedTemporaryFile(suffix=".zip", delete=False) as tmp: tmp_path = tmp.name - with zipfile.ZipFile(tmp_path, "w", zipfile.ZIP_DEFLATED) as archive: + with zipfile.ZipFile(tmp_path, "w", zipfile.ZIP_STORED) as archive: # Save model state dict state_dict_buffer = io.BytesIO() torch.save(self.state_dict(), state_dict_buffer)