chore: add gitignore and small change

2024-09-11 09:39:52 +08:00 · 2024-09-11 09:39:52 +08:00 · 6f1af13e03
parent ccbea71b65
commit 6f1af13e03
2 changed files with 2 additions and 1 deletions
--- a/.gitignore
+++ b/.gitignore
@ -167,3 +167,4 @@ config/
 saves/
 output/
 wandb/
+.vscode
--- 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"