diff --git a/run_once.sh b/run_once.sh
index cf13de0b..ce85dc49 100644
--- a/run_once.sh
+++ b/run_once.sh
@@ -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