Update pipeline.py

pipeline.py

@@ -1015,7 +1015,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                     context_group = []
                     for i in range(context_size):
                         # calculate the frame index
-                        frame_index = ((t+1) * context_group_index * (context_size-overlap)) + i
+                        frame_index = ((t+1) * context_group_index * (context_size-overlap)) + (offset * t) + i
                         # wrap around at the end
                         if frame_index >= num_frames:frame_index = frame_index % num_frames
                         context_group.append(frame_index)
@@ -1023,7 +1023,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                 context_indexes.append(context_groups)
             return context_indexes
                 
-        context_indexes = context_scheduler(context_size, overlap, num_frames, len(timesteps))
+        context_indexes = context_scheduler(context_size, overlap, step, num_frames, len(timesteps))
         
         print(f"Context indexes: {context_indexes}")