fixes

2024-02-29 15:17:39 +02:00
parent 22e0e65f3b
commit 07906ead59
3 changed files with 4 additions and 5 deletions
@@ -894,8 +894,6 @@ 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 * 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
@@ -938,7 +936,6 @@ class VAEHook:
                        tile = task[1](tile)
                        #print(tiles[i].shape, tile.shape, task)
                    pbar.update(1)
-                    pbar_comfy.update(1)

                if interrupted: break