Update pipeline_minimax_remover.py

pipeline_minimax_remover.py

@@ -121,6 +121,7 @@ class Minimax_Remover_Pipeline(DiffusionPipeline):
         output_type: Optional[str] = "np",
         iterations: int = 16
     ):
+
         self._current_timestep = None
         self._interrupt = False
         device = self._execution_device
@@ -145,16 +146,11 @@ 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)
 
@@ -199,4 +195,4 @@ class Minimax_Remover_Pipeline(DiffusionPipeline):
         video = self.vae.decode(latents, return_dict=False)[0]
         video = self.video_processor.postprocess_video(video, output_type=output_type)
 
-        return WanPipelineOutput(frames=video)
+        return WanPipelineOutput(frames=video)