Update pipeline.py
Browse files- pipeline.py +4 -0
pipeline.py
CHANGED
|
@@ -972,8 +972,12 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
| 972 |
if context_group == 0:current_context_start = 0
|
| 973 |
else:current_context_start = context_group * (context_size - overlap)
|
| 974 |
|
|
|
|
|
|
|
|
|
|
| 975 |
# select the relevent context from the latents
|
| 976 |
current_context_latents = latents[:, :, current_context_start : current_context_start + context_size, :, :]
|
|
|
|
| 977 |
|
| 978 |
|
| 979 |
# for context_group in range(num_context_groups):
|
|
|
|
| 972 |
if context_group == 0:current_context_start = 0
|
| 973 |
else:current_context_start = context_group * (context_size - overlap)
|
| 974 |
|
| 975 |
+
# print current start, total frames, and context size, and if end frame is greater than total frames
|
| 976 |
+
print(f"Current context start: {current_context_start}, total frames: {num_frames}, context size: {context_size}, end frame: {current_context_start + context_size}")
|
| 977 |
+
|
| 978 |
# select the relevent context from the latents
|
| 979 |
current_context_latents = latents[:, :, current_context_start : current_context_start + context_size, :, :]
|
| 980 |
+
# if context_start + context_size > num_frames:
|
| 981 |
|
| 982 |
|
| 983 |
# for context_group in range(num_context_groups):
|