Skip to content

Commit 5537bbc

Browse files
Update train.py
1 parent eeff56e commit 5537bbc

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

autotest/module/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def get_cmd(config):
4141

4242
if train_type == "sft":
4343
command = (
44-
f"sleep inf; cd {current_dir}; pwd; {pip_package}; export GITHUB_RUN_ID={config.get('run_id')}; export WORK_DIR={work_dir}; "
44+
f"cd {current_dir}; pwd; {pip_package}; export GITHUB_RUN_ID={config.get('run_id')}; export WORK_DIR={work_dir}; "
4545
+ cudnn_patch
4646
+ f"torchrun --nproc-per-node {nproc_per_node} --master_addr=${{MASTER_ADDR}} --master_port=${{MASTER_PORT}} --nnodes=${{WORLD_SIZE}} --node_rank=${{RANK}} "
4747
+ f"xtuner/v1/train/cli/{train_type}.py"

0 commit comments

Comments
 (0)