Skip to content

Commit 5a26243

Browse files
authored
fix: correct CONFIG_FILE path in multinode_trainer script (#1374)
1 parent 01f4e50 commit 5a26243

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

multinode_trainer.slurm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ export NCCL_BUFFSIZE=2097152
5454
#export TORCH_DIST_INIT_BARRIER=1
5555
export FI_EFA_SET_CUDA_SYNC_MEMOPS=0
5656

57-
CONFIG_FILE=${CONFIG_FILE:-"./torchtitan/models/llama/train_configs/llama3_8b.toml"}
57+
CONFIG_FILE=${CONFIG_FILE:-"./torchtitan/models/llama3/train_configs/llama3_8b.toml"}
5858

5959
dcgmi profile --pause
6060
# adjust sbatch --ntasks and sbatch --nodes above and --nnodes below

0 commit comments

Comments
 (0)