Merge branch 'main' into feature/train-babel

sotopia-lab · Dec 21, 2023 · e464a1c · e464a1c
2 parents 98205ed + 388e761
commit e464a1c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/llm_self_train/resources/train_args.yml b/llm_self_train/resources/train_args.yml
@@ -29,7 +29,8 @@ overwrite_cache: true
 output_dir: NEED-CONFIG
 overwrite_output_dir: true
 logging_steps: 1
-save_strategy: "epoch"
+save_strategy: "steps"
+save_steps: 1000000000
 save_total_limit: 5
 use_auth_token: True
 wandb_token: NEED-CONFIG

diff --git a/llm_self_train/train.py b/llm_self_train/train.py
@@ -9,6 +9,7 @@
 with open('config.yml', 'r') as file:
     config = yaml.safe_load(file)
 
+
 def should_stop(run_sft_completed):
     return run_sft_completed.value