From e754b62ccd2b23149d5ac5942e48d443c059897a Mon Sep 17 00:00:00 2001 From: wql Date: Thu, 5 Sep 2024 15:28:27 +0800 Subject: [PATCH] fix: test --- prepare_yaml_file.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py index 36357f6b..7428f229 100644 --- a/prepare_yaml_file.py +++ b/prepare_yaml_file.py @@ -28,7 +28,7 @@ def main(): model_name_or_path = "../../models/llama-2-7b-ms" template = "llama2" elif model == "Qwen-7B": - model_name_or_path = "../../models/Qwen-7B" + model_name_or_path = "/root/models/Qwen-7B" template = "qwen" else: print("ERROR: model not supported.")