Merge branch 'examples/tp-ckpt' of github.com:Lightning-AI/lightning …

…into examples/tp-ckpt
Lightning-AI · May 7, 2024 · 5d28cf9 · 5d28cf9
2 parents 0665b5a + bf10e24
commit 5d28cf9
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/lightning/fabric/strategies/model_parallel.py b/src/lightning/fabric/strategies/model_parallel.py
@@ -424,7 +424,6 @@ def _load_checkpoint(
     rank: int,
     strict: bool = True,
 ) -> Dict[str, Any]:
-
     from torch.distributed.checkpoint.state_dict import (
         StateDictOptions,
         get_model_state_dict,