Update pipeline.py

pipeline.py

@@ -518,9 +518,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                 )
 
     # Copied from diffusers.pipelines.text_to_video_synthesis.pipeline_text_to_video_synth.TextToVideoSDPipeline.prepare_latents
-    def prepare_latents(
-        self, batch_size, num_channels_latents, num_frames, height, width, dtype, device, generator, latents=None, smooth_weight=0.5
-    ):
+    def prepare_latents(self, batch_size, num_channels_latents, num_frames, height, width, dtype, device, generator, latents=None):
         shape = (
             batch_size,
             num_channels_latents,
@@ -543,9 +541,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
         latents = latents * self.scheduler.init_noise_sigma
         return latents
 
-    def prepare_latents_consistent(
-        self, batch_size, num_channels_latents, num_frames, height, width, dtype, device, generator, latents=None
-    ):
+    def prepare_latents_consistent(self, batch_size, num_channels_latents, num_frames, height, width, dtype, device, generator, latents=None,smooth_weight=0.5,smooth_steps=3):
         shape = (
             batch_size,
             num_channels_latents,
@@ -563,7 +559,6 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
             latents = randn_tensor(shape, generator=generator, device=device, dtype=dtype)
 
             # blend each frame with the surrounding N frames making sure to wrap around at the end
-            smooth_steps = 3
             for i in range(num_frames):
                 blended_latent = torch.zeros_like(latents[:, :, i])
                 for s in range(-smooth_steps, smooth_steps + 1):
@@ -817,6 +812,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
         init_image_strength: Optional[float] = 1.0,
         init_noise_correlation: Optional[float] = 0.0,
         latent_mode: Optional[str] = "normal",
+        smooth_weight: Optional[float] = 0.5,
+        smooth_steps: Optional[int] = 3,
     ):
         r"""
         The call function to the pipeline for generation.
@@ -997,6 +994,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                 device,
                 generator,
                 latents,
+                smooth_weight,
+                smooth_steps,
             )