diff --git a/examples/slurm/submit_multinode.sh b/examples/slurm/submit_multinode.sh index 3fc782e8cb4..b724d2a1955 100644 --- a/examples/slurm/submit_multinode.sh +++ b/examples/slurm/submit_multinode.sh @@ -38,5 +38,5 @@ export SCRIPT_ARGS=" \ " # This step is necessary because accelerate launch does not handle multiline arguments properly -export CMD="$LAUNCHER $PYTHON_FILE $ARGS" -srun $CMD \ No newline at end of file +export CMD="$LAUNCHER $SCRIPT $SCRIPT_ARGS" +srun $CMD