Skip to content

Commit

Permalink
fix : CONFIG_FILE path (#905)
Browse files Browse the repository at this point in the history
fix file path './train_configs/llama3_8b.toml' to
'./torchtitan/models/llama/train_configs/llama3_8b.tom'
  • Loading branch information
kyoungbinkim authored Feb 28, 2025
1 parent f7fb0cf commit 31007c3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion multinode_trainer.slurm
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ export NCCL_BUFFSIZE=2097152
#export TORCH_DIST_INIT_BARRIER=1
export FI_EFA_SET_CUDA_SYNC_MEMOPS=0

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

dcgmi profile --pause
# adjust sbatch --ntasks and sbatch --nodes above and --nnodes below
Expand Down

0 comments on commit 31007c3

Please sign in to comment.