args for finetuning.py

1 files changed, 1 insertions, 1 deletions

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(