Update pipeline.py

pipeline.py

@@ -1020,14 +1020,17 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                     # Initialize a list to store context indexes for this group
                     context_group_indexes = []
                     # Iterate over each index in the context group
+                    local_context_size = context_size
+                    if timestep <= 1:
+                        local_context_size = context_size * 2
                     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 <= 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
+                        # 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 * (local_context_size - overlap)) + (offset * timestep) + index
                         # If frame index exceeds total frames, wrap around
                         if frame_index >= total_frames:
                             frame_index %= total_frames