Update pipeline.py

pipeline.py

@@ -1058,6 +1058,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                             noise_pred_text = noise_pred_text[:, :, :-wrap_count, :, :]
                         noise_pred_uncond_sum[:, :, current_context_start : current_context_start + context_size, :, :] += noise_pred_uncond
                         noise_pred_text_sum[:, :, current_context_start : current_context_start + context_size, :, :] += noise_pred_text
+                        latent_counter[current_context_start : current_context_start + context_size] += 1
                         
                         # print min and max values of noise_pred_uncond_sum and noise_pred_text_sum
                         print(f"noise_pred_uncond_sum min: {noise_pred_uncond_sum.min()} max: {noise_pred_uncond_sum.max()}")
@@ -1069,7 +1070,6 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                 # perform guidance
                 if do_classifier_free_guidance:
                     latent_counter = latent_counter.reshape(1, 1, num_frames, 1, 1)
-                    print(f"latent_counter min: {latent_counter.min()} max: {latent_counter.max()}")
                     noise_pred_uncond = noise_pred_uncond_sum / latent_counter
                     noise_pred_text = noise_pred_text_sum / latent_counter
                     noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)