ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #94 from unslothai/fix/pass-save-steps-to-trainer

Browse source 
Adding save-steps to the SFTConfig
This commit is contained in:
Roland Tannous 2026-02-15 17:25:48 +04:00 • committed by GitHub
commit 6fc52d6535
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
studio/backend/core/training/trainer.py
View file

@ -562,6 +562,12 @@ class UnslothTrainer:
config_args["warmup_steps"] = 5
print(f"Using default warmup_steps: 5\n")
# Add save_steps if specified
save_steps_val = training_args.get('save_steps', 0)
if save_steps_val and save_steps_val > 0:
config_args["save_steps"] = save_steps_val
config_args["save_strategy"] = "steps"
# If max_steps is specified, use it instead of epochs
max_steps_val = training_args.get('max_steps', 0)
if max_steps_val and max_steps_val > 0: