smoothieAI
/

pipeline_animatediff_context

Model card Files Files and versions Community

smoothieAI commited on Jan 9, 2024

Commit

c767690

·

1 Parent(s): b86fb79

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -0

pipeline.py CHANGED Viewed

@@ -768,6 +768,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                 # shuffle rotate latent images by step places, wrapping around the last 2 to the start
                 latents = torch.cat([latents[:, :, -step:, :, :], latents[:, :, :-step, :, :]], dim=2)
         if output_type == "latent":
             return AnimateDiffPipelineOutput(frames=latents)

                 # shuffle rotate latent images by step places, wrapping around the last 2 to the start
                 latents = torch.cat([latents[:, :, -step:, :, :], latents[:, :, :-step, :, :]], dim=2)
+        print("Done denoising")
         if output_type == "latent":
             return AnimateDiffPipelineOutput(frames=latents)