We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2c2ab9e commit 54b859aCopy full SHA for 54b859a
src/lightning/fabric/utilities/cloud_io.py
@@ -56,6 +56,9 @@ def _load(
56
weights_only=weights_only,
57
)
58
if str(path_or_url).startswith("http"):
59
+ if weights_only is None:
60
+ weights_only = True
61
+ log.debug(f"Default to `weights_only=True` for remote checkpoint: {path_or_url}")
62
return torch.hub.load_state_dict_from_url(
63
str(path_or_url),
64
map_location=map_location, # type: ignore[arg-type]
0 commit comments