diff --git a/src/nemo_run/core/execution/slurm.py b/src/nemo_run/core/execution/slurm.py index 5fc522c..a8e74f9 100644 --- a/src/nemo_run/core/execution/slurm.py +++ b/src/nemo_run/core/execution/slurm.py @@ -69,7 +69,8 @@ def your_slurm_executor() -> run.SlurmExecutor: user=os.environ["SLURM_USER"], job_dir=os.environ["SLURM_JOBDIR"], ) - packager = GitArchivePackager(launcher="torchrun") + packager = GitArchivePackager() + launcher = "torchrun" executor = SlurmExecutor( account=os.environ["SLURM_ACCT"], partition=os.environ["SLURM_PARTITION"], @@ -79,6 +80,7 @@ def your_slurm_executor() -> run.SlurmExecutor: container_image=os.environ["BASE_IMAGE"], time="00:30:00", packager=packager, + launcher=launcher, ) return executor