From b85add402c77e5aeeb333a5c09aa22d4a1f3d4a7 Mon Sep 17 00:00:00 2001 From: wql Date: Sat, 12 Oct 2024 11:24:04 +0800 Subject: [PATCH] chore: change models path --- prepare_yaml_file.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py index ce84dbff..329ca772 100644 --- a/prepare_yaml_file.py +++ b/prepare_yaml_file.py @@ -16,19 +16,19 @@ def main(): model_name_or_path = "" template = "" 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" 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" elif model == "ChatGLM2-6B": - model_name_or_path = "../../../models/chatglm2-6b" + model_name_or_path = "/home/ma-user/models/chatglm2-6b" template = "chatglm2" 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" elif model == "Qwen-7B": - model_name_or_path = "../../../models/qwen" + model_name_or_path = "/home/ma-user/models/qwen" template = "qwen" else: print("ERROR: model not supported or model name wrong")