xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-03 20:37:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

check if need to disable pbar

Browse Source
This commit is contained in:
Kohaku-Blueleaf 2025-06-07 05:34:24 +08:00
parent b3b36e5a69
commit 1baa1bdafc
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

@ -500,7 +500,7 @@ class TrainLoraNode:
# Training loop
torch.cuda.empty_cache()
try:
for step in (pbar:=tqdm.trange(steps, desc="Training LoRA", smoothing=0.01)):
for step in (pbar:=tqdm.trange(steps, desc="Training LoRA", smoothing=0.01, disable=not comfy.utils.PROGRESS_BAR_ENABLED)):
# Generate random sigma
sigma = mp.model.model_sampling.percent_to_sigma(
torch.rand((1,)).item()