From 72a0997cdc10b50dfa5998db86221517f006449a Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Wed, 23 Oct 2024 09:49:45 +0800
Subject: [PATCH] chore: add 9g-8B file path

---
 prepare_yaml_file.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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"