From ceb01459feb5cbe53b4c4465ce29c21b1742d760 Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Thu, 5 Sep 2024 13:04:54 +0800
Subject: [PATCH] fix: fix bug

---
 prepare_yaml_file.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py
index 10931229..4c864328 100644
--- a/prepare_yaml_file.py
+++ b/prepare_yaml_file.py
@@ -12,7 +12,9 @@ def main():
         yaml_file = './results/lora_sft_template.yaml'
     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 = ""