Skip to content

Commit 9f709d6

Browse files
committed
default is none
1 parent 2de885c commit 9f709d6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/fabric/strategies/model_parallel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -411,7 +411,7 @@ def _load_checkpoint(
411411
state: dict[str, Union[Module, Optimizer, Any]],
412412
strict: bool = True,
413413
optimizer_states_from_list: bool = False,
414-
weights_only: bool = False,
414+
weights_only: Optional[bool] = None,
415415
) -> dict[str, Any]:
416416
from torch.distributed.checkpoint.state_dict import (
417417
StateDictOptions,

0 commit comments

Comments
 (0)