From b83275d5c6f03313c096d5e42fcae997249980f3 Mon Sep 17 00:00:00 2001 From: wql Date: Thu, 19 Sep 2024 13:52:25 +0800 Subject: [PATCH] chore: change models path --- prepare_yaml_file.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py index 9c906eb1..bc2c7e8d 100644 --- a/prepare_yaml_file.py +++ b/prepare_yaml_file.py @@ -16,16 +16,16 @@ def main(): model_name_or_path = "" template = "" if model == "9g-8B": - model_name_or_path = "ms_cache/hub/qy/sft_8b_v2" + model_name_or_path = "../../../models/sft_8b_v2" template = "default" elif model == "Baichuan2-7B": - model_name_or_path = "ms_cache/hub/baichuan-inc/Baichuan2-7B-Base" + model_name_or_path = "../../../models/Baichuan2-7B-Base" template = "baichuan2" elif model == "ChatGLM2-6B": - model_name_or_path = "ms_cache/hub/zhipuai/chatglm2-6b" + model_name_or_path = "../../../models/chatglm2-6b" template = "chatglm2" elif model == "Llama2-7B": - model_name_or_path = "ms_cache/hub/modelscope/llama-2-7b-ms" + model_name_or_path = "../../../models/llama-2-7b-ms" template = "llama2" elif model == "Qwen-7B": model_name_or_path = "../../../models/qwen"