From d330bb2a37c1465a6d321d7a1806253bd2c99bac Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Tue, 9 Dec 2025 23:06:53 +0800
Subject: [PATCH] fix wrong ui pbar update

---
 comfy_extras/nodes_train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy_extras/nodes_train.py b/comfy_extras/nodes_train.py
index dfbe2c1ce..88bc8c8e8 100644
--- a/comfy_extras/nodes_train.py
+++ b/comfy_extras/nodes_train.py
@@ -341,7 +341,7 @@ class TrainSampler(comfy.samplers.Sampler):
             if (i + 1) % self.grad_acc == 0:
                 self.optimizer.step()
                 self.optimizer.zero_grad()
-                ui_pbar.update(1)
+            ui_pbar.update(1)
         torch.cuda.empty_cache()
         return torch.zeros_like(latent_image)