wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-17 18:13:01 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

fix wrong ui pbar update

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2025-12-09 23:06:53 +08:00
parent 305602c668
commit d330bb2a37
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_train.py
View File

@ -341,7 +341,7 @@ class TrainSampler(comfy.samplers.Sampler):
if (i + 1) % self.grad_acc == 0: if (i + 1) % self.grad_acc == 0:
self.optimizer.step() self.optimizer.step()
self.optimizer.zero_grad() self.optimizer.zero_grad()
ui_pbar.update(1) ui_pbar.update(1)
torch.cuda.empty_cache() torch.cuda.empty_cache()
return torch.zeros_like(latent_image) return torch.zeros_like(latent_image)