Update pipeline.py

pipeline.py

@@ -1528,10 +1528,10 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                         torch.cuda.synchronize()  # Synchronize GPU before batch addition
                         time_batch_addition_start = time.time()
 
-                        for i, context_index in enumerate(current_context_indexes):
+                        for z, context_index in enumerate(current_context_indexes):
                             # Perform batch addition
-                            noise_pred_uncond_sum[..., context_index, :, :] += noise_pred_uncond[:, :, i, :, :]
-                            noise_pred_text_sum[..., context_index, :, :] += noise_pred_text[:, :, i, :, :]
+                            noise_pred_uncond_sum[..., context_index, :, :] += noise_pred_uncond[:, :, z, :, :]
+                            noise_pred_text_sum[..., context_index, :, :] += noise_pred_text[:, :, z, :, :]
 
                             # Batch increment for latent_counter
                             latent_counter[context_index] += 1