Update pipeline_minimax_remover.py

pipeline_minimax_remover.py

@@ -121,8 +121,6 @@ class Minimax_Remover_Pipeline(DiffusionPipeline):
         output_type: Optional[str] = "np",
         iterations: int = 16
     ):
-        
-        print(f"images.shape: {images.shape}, masks.shape:{masks.shape}")
         self._current_timestep = None
         self._interrupt = False
         device = self._execution_device
@@ -147,11 +145,16 @@ class Minimax_Remover_Pipeline(DiffusionPipeline):
             latents,
         )
 
+        print(f"1 images.shape: {images.shape}, masks.shape:{masks.shape}")
         masks = self.expand_masks(masks, iterations)
+        print(f"2 images.shape: {images.shape}, masks.shape:{masks.shape}")
         masks = self.resize(masks, height, width).to("cuda:0").half()
+        print(f"3 images.shape: {images.shape}, masks.shape:{masks.shape}")
         masks[masks>0] = 1
         images = rearrange(images, "f h w c -> c f h w")
+        print(f"4 images.shape: {images.shape}, masks.shape:{masks.shape}")
         images = self.resize(images[None,...], height, width).to("cuda:0").half()
+        print(f"5 images.shape: {images.shape}, masks.shape:{masks.shape}")
 
         masked_images = images * (1-masks)