fix: fix bug

2024-09-05 13:04:54 +08:00 · 2024-09-05 13:04:54 +08:00 · ceb01459fe
parent cc99691cf4
commit ceb01459fe
1 changed files with 3 additions and 1 deletions
--- a/prepare_yaml_file.py
+++ b/prepare_yaml_file.py
@ -13,6 +13,8 @@ def main():
    elif run_type == "inference":
        yaml_file = './results/predict_template.yaml'
    
+    model_name_or_path = ""
+    template = ""
    if model == "9g-8B":
        model_name_or_path = "../../models/sft_8b_v2"
        template = ""