diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py
index f5b8abc0..36357f6b 100644
--- a/prepare_yaml_file.py
+++ b/prepare_yaml_file.py
@@ -19,7 +19,7 @@ def main():
         model_name_or_path = "../../models/sft_8b_v2"
         template = ""
     elif model == "Baichuan2-7B":
-        model_name_or_path = "../../models/Baichuan2-7B"
+        model_name_or_path = "../../models/Baichuan2-7B-Base"
         template = "baichuan2"
     elif model == "ChatGLM2-6B":
         model_name_or_path = "../../models/chatglm2-6b"