Update on "Prevent legacy_load when weights_only=True (correctly)"

mikaylagawarecki · mikaylagawarecki · commit b829da8ac96c · 2025-01-16T16:44:54.000-08:00
Only prevent `legacy_load` (.tar format removed in #713), not the whole of `_legacy_load` (.tar format + _use_new_zipfile_serialization=False) Differential Revision: [D68301405](https://our.internmc.facebook.com/intern/diff/D68301405) [ghstack-poisoned]
diff --git a/torch/serialization.py b/torch/serialization.py
@@ -1738,9 +1738,7 @@ def persistent_load(saved_id):
         # legacy_load requires that f has fileno()
         # only if offset is zero we can attempt the legacy tar file loader
         try:
-            with closing(
-                tarfile.open(fileobj=f, mode="r:", format=tarfile.PAX_FORMAT)
-            ):
+            with closing(tarfile.open(fileobj=f, mode="r:", format=tarfile.PAX_FORMAT)):
                 if pickle_module is _weights_only_unpickler:
                     raise RuntimeError(
                         "Cannot use ``weights_only=True`` with files saved in the "
