diff --git a/SUPIR/utils/tilevae.py b/SUPIR/utils/tilevae.py
index 97974cc..d6cf4d8 100644
--- a/SUPIR/utils/tilevae.py
+++ b/SUPIR/utils/tilevae.py
@@ -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
 
diff --git a/nodes.py b/nodes.py
index 2ac09ea..b6af10d 100644
--- a/nodes.py
+++ b/nodes.py
@@ -130,6 +130,7 @@ class SUPIR_Upscale:
                                                 cfg_scale_start=cfg_scale_start, control_scale_start=control_scale_start)
                 
                 out.append(samples.squeeze(0).cpu())
+                print("Sampled image ", i, " out of ", B)
                 pbar.update(1)
             if not keep_model_loaded:
                     self.model = None
diff --git a/requirements.txt b/requirements.txt
index 9842c42..08cab79 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,5 +8,6 @@ open-clip-torch>=2.17.1
 opencv-python>=4.7.0.72
 pandas>=2.0.1
 Pillow>=9.4.0
-pytorch-lightning>=2.1.2
-PyYAML>=6.0
\ No newline at end of file
+pytorch-lightning==2.1.2
+PyYAML>=6.0
+omegaconf
\ No newline at end of file