Apply isort and black reformatting

Signed-off-by: JRD971000 <[email protected]>
NVIDIA · Jul 15, 2024 · cbaac3f · cbaac3f
1 parent d34932b
commit cbaac3f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/examples/nlp/language_modeling/mamba_change_num_partition.py b/examples/nlp/language_modeling/mamba_change_num_partition.py
@@ -851,7 +851,6 @@ def main():
         model.cfg.global_batch_size = None
         model.cfg.micro_batch_size = None
 
-
         model = MegatronMambaModel(model.cfg, trainer)
         model = model.to('cpu')
         model._save_restore_connector = NLPSaveRestoreConnector()