Update pipeline_minimax_remover.py

pipeline_minimax_remover.py

@@ -121,7 +121,8 @@ 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
@@ -151,7 +152,6 @@ class Minimax_Remover_Pipeline(DiffusionPipeline):
         masks[masks>0] = 1
         images = rearrange(images, "f h w c -> c f h w")
         images = self.resize(images[None,...], height, width).to("cuda:0").half()
-        print(f"images.shape: {images.shape}, masks.shape:{masks.shape}")
 
         masked_images = images * (1-masks)