fix: fix typo

This commit is contained in:
wql 2024-09-05 15:17:39 +08:00
parent 5baa46a798
commit 3d018c8248
1 changed files with 2 additions and 2 deletions

View File

@ -35,11 +35,11 @@ export USE_MODELSCOPE_HUB=1
# echo "${gpu_status_pid}"
if [ "${gpu_cnt}"="1" ]; then
ASCEND_RT_VISIBLE_DEVICES=0 llamafactory-cli train ${output_dir}/${run_name}.yml | tee "${output_dir}/log.txt" &
ASCEND_RT_VISIBLE_DEVICES=0 llamafactory-cli train ${output_dir}/${run_name}.yaml | tee "${output_dir}/log.txt" &
train_pid=$!
echo "Start train"
else
FORCE_TORCHRUN=1 llamafactory-cli train ${output_dir}/${run_name}.yml | tee "${output_dir}/log.txt" &
FORCE_TORCHRUN=1 llamafactory-cli train ${output_dir}/${run_name}.yaml | tee "${output_dir}/log.txt" &
train_pid=$!
echo "Start train"
fi