edited merge conflict

NVIDIA · Oct 24, 2024 · accc256 · accc256
1 parent 302afb7
commit accc256
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/lightning/io/connector.py b/nemo/lightning/io/connector.py
@@ -184,7 +184,7 @@ def nemo_save(self, output_path: Path, trainer: pl.Trainer, dump_io: bool = True
         trainer.strategy.setup(trainer)
         output_path = Path(output_path)
         output_path.mkdir(parents=True, exist_ok=True)
-        trainer.save_checkpoint(ckpt_to_weights_subdir(output_path))
+        trainer.save_checkpoint(output_path)
         if getattr(trainer.strategy, "async_save", False):
             trainer.strategy.checkpoint_io.maybe_finalize_save_checkpoint(blocking=True)