tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

add save_only_model option (#1634)

Browse Source
This commit is contained in:
Jeffrey Quesnelle
2024-05-16 21:23:18 -07:00
committed by GitHub
parent 891ae8aa13
commit 702a669cad
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/core/trainer_builder.py
View File

@@ -1123,6 +1123,8 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
# default to saving each epoch if not defined # default to saving each epoch if not defined
training_arguments_kwargs["save_strategy"] = "epoch" training_arguments_kwargs["save_strategy"] = "epoch"
training_arguments_kwargs["save_only_model"] = self.cfg.save_only_model
if self.cfg.do_bench_eval: if self.cfg.do_bench_eval:
training_arguments_kwargs["do_bench_eval"] = self.cfg.do_bench_eval training_arguments_kwargs["do_bench_eval"] = self.cfg.do_bench_eval
if self.cfg.bench_dataset: if self.cfg.bench_dataset:

1
src/axolotl/utils/config/models/input/v0_4_1/__init__.py
View File

@@ -574,6 +574,7 @@ class AxolotlInputConfig(
logging_steps: Optional[int] = None logging_steps: Optional[int] = None
early_stopping_patience: Optional[int] = None early_stopping_patience: Optional[int] = None
load_best_model_at_end: Optional[bool] = False load_best_model_at_end: Optional[bool] = False
save_only_model: Optional[bool] = False
neftune_noise_alpha: Optional[float] = None neftune_noise_alpha: Optional[float] = None