From 96e8728fea7b6bfad93e70dc688ab49464fe8465 Mon Sep 17 00:00:00 2001
From: pks <pks@pks.rocks>
Date: Sun, 30 Nov 2025 21:06:48 +0100
Subject: args for finetuning.py

---
 finetuning.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'finetuning.py')

diff --git a/finetuning.py b/finetuning.py
index 161fb54..abb4b89 100755
--- a/finetuning.py
+++ b/finetuning.py
@@ -138,7 +138,7 @@ def main():
         optim="adamw_torch_8bit",  # Alternative from BnB: paged_adamw_8bit
         remove_unused_columns=False,
         logging_steps=args.logging_steps,
-        save_strategy="epochs",
+        save_strategy="epoch",
     )
 
     trainer = SFTTrainer(
-- 
cgit v1.2.3