From d89b11715e01be4966da07a9af3e8f3f2f187b53 Mon Sep 17 00:00:00 2001 From: Alex Butler Date: Sat, 15 Aug 2026 19:01:34 +0100 Subject: [PATCH] Add suggestions --- comfy/sd.py | 4 ++-- comfy/utils.py | 5 +++++ 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/comfy/sd.py b/comfy/sd.py index 5d566bd61..d00fdb73f 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -1412,9 +1412,9 @@ class VAE: with model_management.cuda_device_context(self.device): if dims == 1: args.pop("tile_y") - samples = self.encode_tiled_1d(pixel_samples, **args) + samples = self.encode_tiled_1d(pixel_samples, **args, term_pbar_desc=term_pbar_desc) elif dims == 2: - samples = self.encode_tiled_(pixel_samples, **args) + samples = self.encode_tiled_(pixel_samples, **args, term_pbar_desc=term_pbar_desc) elif dims == 3: if self.handles_tiling: samples = self._encode_tiled_owned(pixel_samples, **self._owned_tiled_args(tile_x, tile_y, overlap, tile_t, overlap_t)) diff --git a/comfy/utils.py b/comfy/utils.py index fe321afab..6f1adeb20 100644 --- a/comfy/utils.py +++ b/comfy/utils.py @@ -1171,9 +1171,13 @@ def tiled_scale_multidim(samples, function, tile=(64, 64), overlap=8, upscale_am # handle entire input fitting in a single tile if all(s.shape[d+2] <= tile[d] for d in range(dims)): + if term_pbar_desc and term_pbar is None: + term_pbar = tqdm(desc=term_pbar_desc, total=samples.shape[0]) output[b:b+1] = function(s).to(output_device) if pbar is not None: pbar.update(1) + if term_pbar is not None: + term_pbar.update(1) continue out = output[b:b+1].zero_() @@ -1296,6 +1300,7 @@ class ProgressBar: self.current = value if self.term_pbar and inc > 0: + self.term_pbar.total = self.total self.term_pbar.update(inc) if value >= self.total: self.term_pbar.close()