From 6f967041eda931e778a314857b09ee94af38fa26 Mon Sep 17 00:00:00 2001 From: nocrcl <174792533+nocrcl@users.noreply.github.com> Date: Wed, 24 Sep 2025 14:36:24 +0200 Subject: [PATCH] Update nodes_train.py Removed GUI progressbar as discussed --- comfy_extras/nodes_train.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/comfy_extras/nodes_train.py b/comfy_extras/nodes_train.py index 4fb00fdb6..aebaa7100 100644 --- a/comfy_extras/nodes_train.py +++ b/comfy_extras/nodes_train.py @@ -53,9 +53,7 @@ class TrainSampler(comfy.samplers.Sampler): cond = model_wrap.conds["positive"] dataset_size = sigmas.size(0) torch.cuda.empty_cache() - pbar_gui = comfy.utils.ProgressBar(self.total_steps) for i in (pbar:=tqdm.trange(self.total_steps, desc="Training LoRA", smoothing=0.01, disable=not comfy.utils.PROGRESS_BAR_ENABLED)): - pbar_gui.update(1) noisegen = comfy_extras.nodes_custom_sampler.Noise_RandomNoise(self.seed + i * 1000) indicies = torch.randperm(dataset_size)[:self.batch_size].tolist()