fix: fix typo
This commit is contained in:
parent
5baa46a798
commit
3d018c8248
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue