Update pipeline.py

pipeline.py

@@ -1500,13 +1500,14 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                         time_chunk = time.time()
                         noise_pred_uncond, noise_pred_text = torch.chunk(noise_pred, 2, dim=0)
                         print("chunk time", time.time() - time_chunk)
-                        for i,context_index in range(len(current_context_indexes)):
+                        for i, 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[:, :, i, :, :]
+                            noise_pred_text_sum[..., context_index, :, :] += noise_pred_text[:, :, i, :, :]
 
                             # Batch increment for latent_counter
                             latent_counter[context_index] += 1
+
                         
                     print("guidance time", time.time() - start_guidance_time)