chore: up

2025-02-14 02:13:40 +06:00
parent f013e8efe6
commit 45f45f4bdb
3 changed files with 37 additions and 27 deletions
--- a/trainer.py
+++ b/trainer.py
@@ -65,6 +65,9 @@ class CustomTrainer:
            bf16=torch.cuda.is_bf16_supported(),
            optim="adamw_8bit",
            report_to=report_to,
+            save_strategy="steps",
+            save_steps=50,
+            save_total_limit=3,
        )

        return SFTTrainer(