smoothieAI
/

pipeline_animatediff_context_controlnet

Model card Files Files and versions Community

smoothieAI commited on Jan 27, 2024

Commit

62d7836

·

verified ·

1 Parent(s): f187a55

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -1338,7 +1338,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
             latent_timestep = timesteps[:1].repeat(batch_size * num_videos_per_prompt)
             self._num_timesteps = len(timesteps)
             num_channels_latents = self.unet.config.in_channels
-            latents = self.prepare_latents(
                 video=init_image,
                 height=height,
                 width=width,

             latent_timestep = timesteps[:1].repeat(batch_size * num_videos_per_prompt)
             self._num_timesteps = len(timesteps)
             num_channels_latents = self.unet.config.in_channels
+            latents = self.prepare_video_latents(
                 video=init_image,
                 height=height,
                 width=width,