From 36840f031033aac24176662594dac7488627614d Mon Sep 17 00:00:00 2001 From: wql Date: Thu, 5 Sep 2024 14:35:47 +0800 Subject: [PATCH] fix: fix baichuan2 path --- prepare_yaml_file.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"