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

chore: add 9g-8B file path

This commit is contained in:
wql 2024-10-23 09:49:45 +08:00
parent 3c3bc98f5e
commit 72a0997cdc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
prepare_yaml_file.py
View File

@ -16,8 +16,8 @@ def main():
model_name_or_path = "" model_name_or_path = ""
template = "" template = ""
if model == "9g-8B": if model == "9g-8B":
model_name_or_path = "/home/ma-user/models/sft_8b_v2" model_name_or_path = "/home/ma-user/models/8b_sft_model"
template = "default" template = "cpm"
elif model == "Baichuan2-7B": elif model == "Baichuan2-7B":
model_name_or_path = "/home/ma-user/models/Baichuan2-7B-Base" model_name_or_path = "/home/ma-user/models/Baichuan2-7B-Base"
template = "baichuan2" template = "baichuan2"