smoothieAI
/

pipeline_animatediff_context_controlnet

Model card Files Files and versions Community

smoothieAI commited on Jan 28, 2024

Commit

57f4540

·

verified ·

1 Parent(s): c5385e2

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -1531,7 +1531,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                         # Perform batch addition
                         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
                         time_batch_addition_end = time.time()

                         # Perform batch addition
                         noise_pred_uncond_sum[..., current_context_indexes, :, :] += noise_pred_uncond
                         noise_pred_text_sum[..., current_context_indexes, :, :] += noise_pred_text
+                        latent_counter[current_context_indexes] += 1
                         torch.cuda.synchronize()  # Synchronize GPU after batch addition
                         time_batch_addition_end = time.time()