From 6f1af13e0356894a4b9b069b58dc26795a70ba9d Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Wed, 11 Sep 2024 09:39:52 +0800
Subject: [PATCH] chore: add gitignore and small change

---
 .gitignore           | 1 +
 prepare_yaml_file.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 8acdb364..3a00c67b 100644
--- a/.gitignore
+++ b/.gitignore
@@ -167,3 +167,4 @@ config/
 saves/
 output/
 wandb/
+.vscode
diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py
index fb27f279..6cec6715 100644
--- a/prepare_yaml_file.py
+++ b/prepare_yaml_file.py
@@ -17,7 +17,7 @@ def main():
     template = ""
     if model == "9g-8B":
         model_name_or_path = "../../models/sft_8b_v2"
-        template = ""
+        template = "default"
     elif model == "Baichuan2-7B":
         model_name_or_path = "../../models/Baichuan2-7B-Base"
         template = "baichuan2"