sleepyhead111
/

lmt-arr

Model card Files Files and versions

Update scripts/cpt_mt_4b.sh

#1

by fpnBytedance - opened Sep 17, 2025

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

scripts/cpt_mt_4b.sh +2 -1

scripts/cpt_mt_4b.sh CHANGED Viewed

@@ -44,7 +44,8 @@ cp $0 $output_dir
 #     --resume_from_checkpoint $resume_from_checkpoint \
-swift pt \
     --deepspeed  $config_file \
     --add_version False \
     --check_model False \

 #     --resume_from_checkpoint $resume_from_checkpoint \
+# 多机多卡
+FORCE_TORCHRUN=1 NNODES=$worker_num RANK=$worker_index MASTER_ADDR=$host MASTER_PORT=$port swift pt \
     --deepspeed  $config_file \
     --add_version False \
     --check_model False \