Merge branch 'ashors/nemo-ux-drop-optim-states-async' of github.com:N…

…VIDIA/NeMo into ashors/nemo-ux-drop-optim-states-async
NVIDIA · Dec 2, 2024 · 30954ab · 30954ab
2 parents f1c352a + ee523a3
commit 30954ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/lightning/io/pl.py b/nemo/lightning/io/pl.py
@@ -307,7 +307,7 @@ def drop_optimizer_states(self, path):
         from megatron.core import dist_checkpointing
 
         dist_checkpointing.remove_sharded_tensors(path, key_prefix="optimizer")
-    
+
     def adjust_non_strict_load(self, path: _PATH, sharded_state_dict: Dict[str, Any]):
         from megatron.core import dist_checkpointing
         from megatron.core.dist_checkpointing.dict_utils import extract_matching_values