fix: test fix

2024-09-05 15:15:37 +08:00 · 2024-09-05 15:15:37 +08:00 · 5baa46a798
parent 62a486dfc0
commit 5baa46a798
1 changed files with 1 additions and 1 deletions
--- 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 = "'../../models/Qwen-7B'"
        template = "qwen"
    else:
        print("ERROR: model not supported.")