Update pipeline.py

pipeline.py

@@ -1022,9 +1022,9 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                     # Iterate over each index in the context group
                     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
+                        if timestep <= 1:
+                            step_size = (total_frames // context_size)+2
+                            frame_index = ((index * step_size)+group_index)+timestep
                         else:
                             # Calculate the frame index
                             frame_index = (group_index * (context_size - overlap)) + (offset * timestep) + index