diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py index d68cbee7..9c906eb1 100644 --- a/prepare_yaml_file.py +++ b/prepare_yaml_file.py @@ -28,7 +28,7 @@ def main(): model_name_or_path = "ms_cache/hub/modelscope/llama-2-7b-ms" template = "llama2" elif model == "Qwen-7B": - model_name_or_path = "ms_cache/hub/qwen/Qwen-7B" + model_name_or_path = "../../../models/qwen" template = "qwen" else: print("ERROR: model not supported.")