chore: add 9g-8B file path

2024-10-23 09:49:45 +08:00 · 2024-10-23 09:49:45 +08:00 · 72a0997cdc
parent 3c3bc98f5e
commit 72a0997cdc
1 changed files with 2 additions and 2 deletions
--- 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"