Skip to content

Commit ea7d521

Browse files
Removed zip compression (#1210)
Signed-off-by: Charlelie Laurent <[email protected]> Co-authored-by: Corey adams <[email protected]>
1 parent 69cf158 commit ea7d521

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

physicsnemo/models/module.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -541,7 +541,7 @@ def _save_process(module, args, metadata, mod_prefix="") -> None:
541541
with tempfile.NamedTemporaryFile(suffix=".zip", delete=False) as tmp:
542542
tmp_path = tmp.name
543543

544-
with zipfile.ZipFile(tmp_path, "w", zipfile.ZIP_DEFLATED) as archive:
544+
with zipfile.ZipFile(tmp_path, "w", zipfile.ZIP_STORED) as archive:
545545
# Save model state dict
546546
state_dict_buffer = io.BytesIO()
547547
torch.save(self.state_dict(), state_dict_buffer)

0 commit comments

Comments
 (0)