diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py
index 329ca772..0a3d4255 100644
--- a/prepare_yaml_file.py
+++ b/prepare_yaml_file.py
@@ -16,8 +16,8 @@ def main():
     model_name_or_path = ""
     template = ""
     if model == "9g-8B":
-        model_name_or_path = "/home/ma-user/models/sft_8b_v2"
-        template = "default"
+        model_name_or_path = "/home/ma-user/models/8b_sft_model"
+        template = "cpm"
     elif model == "Baichuan2-7B":
         model_name_or_path = "/home/ma-user/models/Baichuan2-7B-Base"
         template = "baichuan2"