Update pipeline.py
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
|
@@ -979,6 +979,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
| 979 |
current_context_latents = latents[:, :, current_context_start : current_context_start + context_size, :, :]
|
| 980 |
# if context_start + context_size > num_frames: append the remaining frames from the start of the latents
|
| 981 |
if current_context_start + context_size > num_frames:
|
|
|
|
| 982 |
current_context_latents = torch.cat([current_context_latents, latents[:, :, :max(current_context_start + context_size - num_frames, 0), :, :]], dim=2)
|
| 983 |
|
| 984 |
|
|
|
|
| 979 |
current_context_latents = latents[:, :, current_context_start : current_context_start + context_size, :, :]
|
| 980 |
# if context_start + context_size > num_frames: append the remaining frames from the start of the latents
|
| 981 |
if current_context_start + context_size > num_frames:
|
| 982 |
+
print(f"Appending {max(current_context_start + context_size - num_frames, 0)} frames from the start of the latents")
|
| 983 |
current_context_latents = torch.cat([current_context_latents, latents[:, :, :max(current_context_start + context_size - num_frames, 0), :, :]], dim=2)
|
| 984 |
|
| 985 |
|