Update pipeline.py

pipeline.py

@@ -1186,6 +1186,9 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
         if do_classifier_free_guidance:
             # concatenate negative prompt embeddings with prompt embeddings on a new dimension after the first batch dimension
             prompt_embeds = torch.stack([negative_prompt_embeds, prompt_embeds], dim=1)
+            
+        print("prompt_embeds shape after stacking")
+        print(prompt_embeds.shape)
 
         if ip_adapter_image is not None:
             output_hidden_state = False if isinstance(self.unet.encoder_hid_proj, ImageProjection) else True
@@ -1484,6 +1487,9 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                     
                     else:
                         # predict the noise residual without contorlnet
+                        # print current context embeding shape
+                        print("current context embeding shape")
+                        print(prompt_embeds[current_prompt_index].shape)
                         noise_pred = self.unet(
                             latent_model_input,
                             t,