Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -1078,7 +1078,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
1078 |
if callback is not None and i % callback_steps == 0:
|
1079 |
callback(i, t, None)
|
1080 |
|
1081 |
-
#
|
1082 |
latents = torch.cat([latents[:, :, -step:, :, :], latents[:, :, :-step, :, :]], dim=2)
|
1083 |
|
1084 |
if output_type == "latent":
|
|
|
1078 |
if callback is not None and i % callback_steps == 0:
|
1079 |
callback(i, t, None)
|
1080 |
|
1081 |
+
# offset latent images by step places, wrapping around the last frames to the start
|
1082 |
latents = torch.cat([latents[:, :, -step:, :, :], latents[:, :, :-step, :, :]], dim=2)
|
1083 |
|
1084 |
if output_type == "latent":
|