From de50af6283c89150a12555e1f52ba4368337d948 Mon Sep 17 00:00:00 2001
From: svlandeg <svlandeg@github.com>
Date: Tue, 30 Dec 2025 09:33:33 +0100
Subject: [PATCH] extend sft loop with one more step

---
 scripts/chat_sft.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/chat_sft.py b/scripts/chat_sft.py
index bbb33e0..5678594 100644
--- a/scripts/chat_sft.py
+++ b/scripts/chat_sft.py
@@ -167,8 +167,8 @@ def get_lr_multiplier(it):
 
 # Go!
 step = 0
-for step in range(num_iterations):
-    last_step = step == num_iterations - 1
+for step in range(num_iterations + 1):
+    last_step = step == num_iterations
 
     # evaluate the validation loss
     if last_step or step % eval_every == 0: