Update pipeline.py
Browse files- pipeline.py +2 -0
pipeline.py
CHANGED
|
@@ -1010,6 +1010,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
| 1010 |
if wrap_count > 0:
|
| 1011 |
# remove the ending frames from current_context_latents
|
| 1012 |
current_context_latents = current_context_latents[:, :, :-wrap_count, :, :]
|
|
|
|
|
|
|
| 1013 |
|
| 1014 |
# compute the previous noisy sample x_t -> x_t-1
|
| 1015 |
current_context_latents = self.scheduler.step(noise_pred, t, current_context_latents, **extra_step_kwargs).prev_sample
|
|
|
|
| 1010 |
if wrap_count > 0:
|
| 1011 |
# remove the ending frames from current_context_latents
|
| 1012 |
current_context_latents = current_context_latents[:, :, :-wrap_count, :, :]
|
| 1013 |
+
# remove the ending frames from noise_pred
|
| 1014 |
+
noise_pred = noise_pred[:, :, :-wrap_count, :, :]
|
| 1015 |
|
| 1016 |
# compute the previous noisy sample x_t -> x_t-1
|
| 1017 |
current_context_latents = self.scheduler.step(noise_pred, t, current_context_latents, **extra_step_kwargs).prev_sample
|