From 379f6d1c47f6112b11ac14ba4052d3adc76bbf2f Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Wed, 25 Sep 2024 10:21:25 +0800
Subject: [PATCH] fix: fix model name

---
 batch_run.sh         | 16 ++++++++--------
 prepare_yaml_file.py |  2 +-
 2 files changed, 9 insertions(+), 9 deletions(-)

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