From ed97c8d9d6796902f1ccdf48829e0bfaebba43d1 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 29 Feb 2024 02:38:19 +0200
Subject: [PATCH] fix pbar

---
 SUPIR/utils/tilevae.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/SUPIR/utils/tilevae.py b/SUPIR/utils/tilevae.py
index dc1e31b..c0375f6 100644
--- a/SUPIR/utils/tilevae.py
+++ b/SUPIR/utils/tilevae.py
@@ -893,7 +893,7 @@ class VAEHook:
         # Task queue execution
         pbar = tqdm(total=num_tiles * len(task_queues[0]), desc=f"[Tiled VAE]: Executing {'Decoder' if is_decoder else 'Encoder'} Task Queue: ")
         import comfy.utils
-        pbar_comfy = comfy.utils.ProgressBar(num_tiles)
+        pbar_comfy = comfy.utils.ProgressBar(num_tiles * len(task_queues[0]))
 
         # execute the task back and forth when switch tiles so that we always
         # keep one tile on the GPU to reduce unnecessary data transfer