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

chore: change models path

This commit is contained in:
wql 2024-10-12 11:24:04 +08:00
parent e72ed9c2ca
commit b85add402c
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
prepare_yaml_file.py
View File

@ -16,19 +16,19 @@ def main():
model_name_or_path = "" model_name_or_path = ""
template = "" template = ""
if model == "9g-8B": if model == "9g-8B":
model_name_or_path = "../../../models/sft_8b_v2" model_name_or_path = "/home/ma-user/models/sft_8b_v2"
template = "default" template = "default"
elif model == "Baichuan2-7B": elif model == "Baichuan2-7B":
model_name_or_path = "../../../models/Baichuan2-7B-Base" model_name_or_path = "/home/ma-user/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 = "/home/ma-user/models/chatglm2-6b"
template = "chatglm2" template = "chatglm2"
elif model == "Llama2-7B": elif model == "Llama2-7B":
model_name_or_path = "../../../models/llama-2-7b-ms" model_name_or_path = "/home/ma-user/models/llama-2-7b-ms"
template = "llama2" template = "llama2"
elif model == "Qwen-7B": elif model == "Qwen-7B":
model_name_or_path = "../../../models/qwen" model_name_or_path = "/home/ma-user/models/qwen"
template = "qwen" template = "qwen"
else: else:
print("ERROR: model not supported or model name wrong") print("ERROR: model not supported or model name wrong")