From d6bc0b3ad659064781a9348896ae9433210793b5 Mon Sep 17 00:00:00 2001
From: Augustin Bussy <augustin.bussy@cscs.ch>
Date: Tue, 8 Jul 2025 18:04:04 +0200
Subject: [PATCH] Updated deprecated Slurm variable name

---
 docs/guides/mlp_tutorials/llm-nanotron-training.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/guides/mlp_tutorials/llm-nanotron-training.md b/docs/guides/mlp_tutorials/llm-nanotron-training.md
index 82c7957f..b45d8fc4 100644
--- a/docs/guides/mlp_tutorials/llm-nanotron-training.md
+++ b/docs/guides/mlp_tutorials/llm-nanotron-training.md
@@ -229,7 +229,7 @@ srun -ul --environment=nanotron bash -c "
    --master-addr=\${MASTER_ADDR} \
    --master-port=\${MASTER_PORT} \
    --nnodes=\${SLURM_NNODES} \
-   --nproc-per-node=\${SLURM_GPUS_PER_TASK} \
+   --nproc-per-node=\${SLURM_GPUS_ON_NODE} \
   \"
 
   torchrun \${TORCHRUN_ARGS} run_train.py --config-file examples/config_tiny_llama_wikitext.yaml