diff --git a/batch_run.sh b/batch_run.sh
index 12420ba1..15561372 100644
--- a/batch_run.sh
+++ b/batch_run.sh
@@ -25,14 +25,14 @@ for ((i=0; i<repeat_cnt; i++)); do
     bash run_once.sh inference Baichuan2-7B 1 500
     bash run_once.sh inference Baichuan2-7B 1 1000
 
-    bash run_once.sh lora_sft chatglm2-6B 8 500
-    bash run_once.sh lora_sft chatglm2-6B 8 1000
-    bash run_once.sh lora_sft chatglm2-6B 1 500
-    bash run_once.sh lora_sft chatglm2-6B 1 1000
-    bash run_once.sh inference chatglm2-6B 8 500
-    bash run_once.sh inference chatglm2-6B 8 1000
-    bash run_once.sh inference chatglm2-6B 1 500
-    bash run_once.sh inference chatglm2-6B 1 1000
+    bash run_once.sh lora_sft ChatGLM2-6B 8 500
+    bash run_once.sh lora_sft ChatGLM2-6B 8 1000
+    bash run_once.sh lora_sft ChatGLM2-6B 1 500s
+    bash run_once.sh lora_sft ChatGLM2-6B 1 1000
+    bash run_once.sh inference ChatGLM2-6B 8 500
+    bash run_once.sh inference ChatGLM2-6B 8 1000
+    bash run_once.sh inference ChatGLM2-6B 1 500
+    bash run_once.sh inference ChatGLM2-6B 1 1000
 
     # bash run_once.sh lora_sft llama-2-7b-ms 8 500
     # bash run_once.sh lora_sft llama-2-7b-ms 8 1000
diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py
index bc2c7e8d..ce84dbff 100644
--- a/prepare_yaml_file.py
+++ b/prepare_yaml_file.py
@@ -31,7 +31,7 @@ def main():
         model_name_or_path = "../../../models/qwen"
         template = "qwen"
     else:
-        print("ERROR: model not supported.")
+        print("ERROR: model not supported or model name wrong")
         sys.exit()
         
     config = None