Update pipeline.py

pipeline.py

@@ -1527,13 +1527,10 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                         # Timing for batch addition and latent counter increment
                         torch.cuda.synchronize()  # Synchronize GPU before batch addition
                         time_batch_addition_start = time.time()
-
-                        current_context_tensor = torch.tensor(current_context_indexes, device=latent_counter.device)
+                        
                         # Perform batch addition
-                        noise_pred_uncond_sum[..., current_context_tensor, :, :] += noise_pred_uncond.transpose(1, 2)
-                        noise_pred_text_sum[..., current_context_tensor, :, :] += noise_pred_text.transpose(1, 2)
-
-                        # Batch increment for latent_counter
+                        noise_pred_uncond_sum[..., current_context_indexes, :, :] += noise_pred_uncond
+                        noise_pred_text_sum[..., current_context_indexes, :, :] += noise_pred_text
                         latent_counter[current_context_tensor] += 1
 
                         torch.cuda.synchronize()  # Synchronize GPU after batch addition