Update src/lightning/fabric/utilities/cloud_io.py

matsumotosan · Borda · web-flow · commit 906e52e13ea5 · 2025-08-27T16:49:45.000-04:00
Co-authored-by: Jirka Borovec &lt;6035284+Borda@users.noreply.github.com&gt;
diff --git a/src/lightning/fabric/utilities/cloud_io.py b/src/lightning/fabric/utilities/cloud_io.py
@@ -56,7 +56,7 @@ def _load(
             weights_only=weights_only,
         )
     if str(path_or_url).startswith("http"):
-        if weights_only is None:
+        if weights_only is None and _TORCH_GREATER_EQUAL_2_6:
             weights_only = True
             log.debug(f"Default to `weights_only=True` for remote checkpoint: {path_or_url}")
         return torch.hub.load_state_dict_from_url(

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ def _load(`
`56`	`56`	`weights_only=weights_only,`
`57`	`57`	`)`
`58`	`58`	`if str(path_or_url).startswith("http"):`
`59`		`- if weights_only is None:`
	`59`	`+ if weights_only is None and _TORCH_GREATER_EQUAL_2_6:`
`60`	`60`	`weights_only = True`
`61`	`61`	log.debug(f"Default to `weights_only=True` for remote checkpoint: {path_or_url}")
`62`	`62`	`return torch.hub.load_state_dict_from_url(`