fix wrong ui pbar update

2026-07-12 23:27:00 +08:00 · 2025-12-09 23:06:53 +08:00 · 2025-12-09 23:06:53 +08:00 · d330bb2a37
commit d330bb2a37
parent 305602c668
1 changed files with 1 additions and 1 deletions
--- 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)