pytorch
diff --git a/‎torch/export/__init__.py‎
Lines changed: 16 additions & 4 deletions b/‎torch/export/__init__.py‎
Lines changed: 16 additions & 4 deletions
@@ -1,3 +1,4 @@
+import logging
 import os
 import warnings
 import zipfile
@@ -52,6 +53,8 @@
 
 PassType = Callable[[torch.fx.GraphModule], Optional[PassResult]]
 
+log: logging.Logger = logging.getLogger(__name__)
+
 
 @deprecated(
     "`torch.export.export_for_training` is deprecated and will be removed in PyTorch 2.10. "
@@ -440,7 +443,8 @@ def load(
             f,
             expected_opset_version=expected_opset_version,
         )
-    except RuntimeError:
+    except RuntimeError as e:
+        log.warning("Ran into the following error when deserializing: %s", e)
         pt2_contents = PT2ArchiveContents({}, {}, {})
 
     if len(pt2_contents.exported_programs) > 0 or len(pt2_contents.extra_files) > 0:
@@ -450,10 +454,18 @@ def load(
         return pt2_contents.exported_programs["model"]
 
     # TODO: For backward compatibility, we support loading a zip file from 2.7. Delete this path in 2.9(?)
-    warnings.warn(
-        "This version of file is deprecated. Please generate a new pt2 saved file."
-    )
     with zipfile.ZipFile(f, "r") as zipf:
+        if "version" not in zipf.namelist():
+            raise RuntimeError(
+                "We ran into an error when deserializing the saved file. "
+                "Please check the warnings above for possible errors. "
+            )
+
+        log.warning(
+            "Trying to deserialize for the older format. This version of file is "
+            "deprecated. Please generate a new pt2 saved file."
+        )
+
         # Check the version
         version = zipf.read("version").decode().split(".")
         from torch._export.serde.schema import (