smoothieAI
/

pipeline_animatediff_context

Model card Files Files and versions Community

smoothieAI commited on Jan 18, 2024

Commit

0678810

·

verified ·

1 Parent(s): a7f4efd

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -1072,7 +1072,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                         # if its the first timestep, spread the indexes out evenly over the full frame range, offset by the group index
                         if timestep <= 1:
                             step_size = 2
-                            # make the context group stretch
                         else:
                             # Calculate the frame index
                             frame_index = (group_index * (local_context_size - overlap)) + (offset * timestep) + index

                         # if its the first timestep, spread the indexes out evenly over the full frame range, offset by the group index
                         if timestep <= 1:
                             step_size = 2
+                            frame_index = (group_index * (local_context_size - overlap)) + (offset * timestep) + index
                         else:
                             # Calculate the frame index
                             frame_index = (group_index * (local_context_size - overlap)) + (offset * timestep) + index