diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py
index 6cec6715..d68cbee7 100644
--- a/prepare_yaml_file.py
+++ b/prepare_yaml_file.py
@@ -16,19 +16,19 @@ def main():
     model_name_or_path = ""
     template = ""
     if model == "9g-8B":
-        model_name_or_path = "../../models/sft_8b_v2"
+        model_name_or_path = "ms_cache/hub/qy/sft_8b_v2"
         template = "default"
     elif model == "Baichuan2-7B":
-        model_name_or_path = "../../models/Baichuan2-7B-Base"
+        model_name_or_path = "ms_cache/hub/baichuan-inc/Baichuan2-7B-Base"
         template = "baichuan2"
     elif model == "ChatGLM2-6B":
-        model_name_or_path = "../../models/chatglm2-6b"
+        model_name_or_path = "ms_cache/hub/zhipuai/chatglm2-6b"
         template = "chatglm2"
     elif model == "Llama2-7B":
-        model_name_or_path = "../../models/llama-2-7b-ms"
+        model_name_or_path = "ms_cache/hub/modelscope/llama-2-7b-ms"
         template = "llama2"
     elif model == "Qwen-7B":
-        model_name_or_path = "../../models/Qwen-7B"
+        model_name_or_path = "ms_cache/hub/qwen/Qwen-7B"
         template = "qwen"
     else:
         print("ERROR: model not supported.")
diff --git a/run_once.sh b/run_once.sh
index 26817027..0a14967a 100644
--- a/run_once.sh
+++ b/run_once.sh
@@ -46,6 +46,6 @@ fi
 wait $train_pid
 echo "Train ended"
 
-sleep 60
+# sleep 60
 kill $npu_status_pid
 echo "Npu status ended"
\ No newline at end of file