Update pipeline.py
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
|
@@ -976,6 +976,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
| 976 |
current_context_latents = latents[:, :, current_context_start : current_context_start + context_size, :, :]
|
| 977 |
|
| 978 |
wrap_count = max(current_context_start + context_size - num_frames, 0)
|
|
|
|
| 979 |
|
| 980 |
# if context_start + context_size > num_frames: append the remaining frames from the start of the latents
|
| 981 |
if wrap_count > 0:
|
|
|
|
| 976 |
current_context_latents = latents[:, :, current_context_start : current_context_start + context_size, :, :]
|
| 977 |
|
| 978 |
wrap_count = max(current_context_start + context_size - num_frames, 0)
|
| 979 |
+
print(f"Wrap count: {wrap_count}")
|
| 980 |
|
| 981 |
# if context_start + context_size > num_frames: append the remaining frames from the start of the latents
|
| 982 |
if wrap_count > 0:
|