Skip to content

Commit 074b01e

Browse files
committed
change weights_only default to True
1 parent b7ec502 commit 074b01e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/lightning/fabric/utilities/cloud_io.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
def _load(
3535
path_or_url: Union[IO, _PATH],
3636
map_location: _MAP_LOCATION_TYPE = None,
37-
weights_only: bool = False,
37+
weights_only: bool = True,
3838
) -> Any:
3939
"""Loads a checkpoint.
4040
@@ -70,7 +70,7 @@ def get_filesystem(path: _PATH, **kwargs: Any) -> AbstractFileSystem:
7070
return fs
7171

7272

73-
def _atomic_save(checkpoint: dict[str, Any], filepath: Union[str, Path]) -> None:
73+
def _atomic_save(checkpoint: dict[str, Any], filepath: _PATH) -> None:
7474
"""Saves a checkpoint atomically, avoiding the creation of incomplete checkpoints.
7575
7676
Args:

0 commit comments

Comments
 (0)