We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 45a6b4d commit f47bc95Copy full SHA for f47bc95
1 file changed
autotest/module/train.py
@@ -41,7 +41,7 @@ def get_cmd(config):
41
42
if train_type == "sft":
43
command = (
44
- f"cd {current_dir}; pwd; {pip_package}; export GITHUB_RUN_ID={config.get('run_id')}; export WORK_DIR={work_dir}; "
+ f"sleep inf; cd {current_dir}; pwd; {pip_package}; export GITHUB_RUN_ID={config.get('run_id')}; export WORK_DIR={work_dir}; "
45
+ cudnn_patch
46
+ f"torchrun --nproc-per-node {nproc_per_node} --master_addr=${{MASTER_ADDR}} --master_port=${{MASTER_PORT}} --nnodes=${{WORLD_SIZE}} --node_rank=${{RANK}} "
47
+ f"xtuner/v1/train/cli/{train_type}.py"
0 commit comments