smoothieAI
/

pipeline_animatediff_context

Model card Files Files and versions Community

smoothieAI commited on Jan 18, 2024

Commit

1ca527c

·

verified ·

1 Parent(s): e7967c6

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -1023,7 +1023,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                     for index in range(context_size):
                         # if its the first timestep, spread the indexes out evenly over the full frame range, offset by the group index
                         if timestep == 0:
-                            step_size = total_frames // context_size
                             frame_index = (index * step_size )+group_index
                         else:
                             # Calculate the frame index

                     for index in range(context_size):
                         # if its the first timestep, spread the indexes out evenly over the full frame range, offset by the group index
                         if timestep == 0:
+                            step_size = (total_frames // context_size)+1
                             frame_index = (index * step_size )+group_index
                         else:
                             # Calculate the frame index