Update pipeline.py

pipeline.py

@@ -862,7 +862,6 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
         return image
         
     @torch.no_grad()
-    # @replace_example_docstring(EXAMPLE_DOC_STRING)
     def __call__(
         self,
         prompt: Union[str, List[str]] = None,
@@ -1278,7 +1277,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                         else:
                             control_model_input = latent_model_input
                             controlnet_prompt_embeds = prompt_embeds
-                        controlnet_prompt_embeds = controlnet_prompt_embeds.repeat_interleave(num_frames, dim=0)
+                        controlnet_prompt_embeds = controlnet_prompt_embeds.repeat_interleave(len(current_context_indexes), dim=0)
 
                         if isinstance(controlnet_keep[i], list):
                             cond_scale = [c * s for c, s in zip(controlnet_conditioning_scale, controlnet_keep[i])]