This commit is contained in:
Huiwenshi
2025-06-14 01:39:07 +08:00
parent db57a4467e
commit 4d67e18386
4 changed files with 11 additions and 4 deletions

View File

@@ -56,7 +56,7 @@ else
fi
NODE_RANK=$node_rank \
HF_HUB_OFFLINE=1 \
HF_HUB_OFFLINE=0 \
MASTER_PORT=12348 \
MASTER_ADDR=$master_ip \
NCCL_SOCKET_IFNAME=bond1 \
@@ -67,4 +67,4 @@ python3 main.py \
--num_gpus 8 \
--config $config \
--output_dir $output_dir \
--deepspeed
--deepspeed