p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: fix baichuan2 path

This commit is contained in:
wql 2024-09-05 14:35:47 +08:00
parent 190fddf27d
commit 36840f0310
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prepare_yaml_file.py
View File

@ -19,7 +19,7 @@ def main():
model_name_or_path = "../../models/sft_8b_v2" model_name_or_path = "../../models/sft_8b_v2"
template = "" template = ""
elif model == "Baichuan2-7B": elif model == "Baichuan2-7B":
model_name_or_path = "../../models/Baichuan2-7B" model_name_or_path = "../../models/Baichuan2-7B-Base"
template = "baichuan2" template = "baichuan2"
elif model == "ChatGLM2-6B": elif model == "ChatGLM2-6B":
model_name_or_path = "../../models/chatglm2-6b" model_name_or_path = "../../models/chatglm2-6b"