Update pipeline.py

pipeline.py

@@ -1290,6 +1290,10 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                         control_model_input = control_model_input.reshape(
                             (-1, control_model_input.shape[2], control_model_input.shape[3], control_model_input.shape[4])
                         )
+                        # print shapes of controlnet inputs and conditions
+                        print("control_model_input", control_model_input.shape)
+                        print("controlnet_prompt_embeds", controlnet_prompt_embeds.shape)
+                        print("current_context_conditioning_frames", current_context_conditioning_frames.shape)
                         
                         down_block_res_samples, mid_block_res_sample = self.controlnet(
                             control_model_input,
@@ -1301,6 +1305,10 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                             return_dict=False,
                         )   
                         
+                        # print shapes of controlnet outputs
+                        print("down_block_res_samples", down_block_res_samples.shape)
+                        print("mid_block_res_sample", mid_block_res_sample.shape)
+                        
                         # predict the noise residual with the added controlnet residuals
                         noise_pred = self.unet(
                             latent_model_input,