Update pipeline.py

pipeline.py

@@ -280,6 +280,8 @@ class Flex2Pipeline(FluxControlPipeline):
                 dtype=self.vae.dtype,
             )
         else:
+            print("inpaint_image.shape",inpaint_image.size)
+            print("inpaint_mask.shape",inpaint_mask.size)
             inpaint_image = self.prepare_image(
                 image=inpaint_image,
                 width=width,
@@ -316,6 +318,10 @@ class Flex2Pipeline(FluxControlPipeline):
         
         # pack the controls
         height_latent_controls, width_latent_controls = latent_controls.shape[2:]
+        print("latent_controls.shape",latent_controls.shape)
+        print("num_control_channels.shape",num_control_channels)
+        print("height_latent_controls",height_latent_controls)
+        print("width_latent_controls",width_latent_controls)
         packed_latent_controls = self._pack_latents(
             latent_controls,
             batch_size * num_images_per_prompt,