Update pipeline.py

pipeline.py

@@ -839,7 +839,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
         guess_mode=False,
     ):
         image = self.control_image_processor.preprocess(image, height=height, width=width).to(dtype=torch.float32)
-        image_batch_size = image.shape[0]
+        # image_batch_size = image.shape[0]
+        image_batch_size = len(image)
         print("prepared control image_batch_size", image_batch_size)
 
         if image_batch_size == 1:
@@ -848,7 +849,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
             # image batch size is the same as prompt batch size
             repeat_by = num_images_per_prompt
 
-        image = image.repeat_interleave(repeat_by, dim=0)
+        # image = image.repeat_interleave(repeat_by, dim=0)
 
         image = image.to(device=device, dtype=dtype)
 
@@ -1285,6 +1286,9 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                             if isinstance(controlnet_cond_scale, list):
                                 controlnet_cond_scale = controlnet_cond_scale[0]
                             cond_scale = controlnet_cond_scale * controlnet_keep[i]
+                            
+                        print("-----------------------")
+                        print("control_model_input.shape", control_model_input.shape)
                                            
                         control_model_input = torch.transpose(control_model_input, 1, 2)
                         control_model_input = control_model_input.reshape(