Update pipeline.py
Browse files- pipeline.py +0 -5
pipeline.py
CHANGED
@@ -1059,10 +1059,6 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
1059 |
noise_pred_uncond_sum[:, :, current_context_start : current_context_start + context_size, :, :] += noise_pred_uncond
|
1060 |
noise_pred_text_sum[:, :, current_context_start : current_context_start + context_size, :, :] += noise_pred_text
|
1061 |
latent_counter[current_context_start : current_context_start + context_size] += 1
|
1062 |
-
|
1063 |
-
# print min and max values of noise_pred_uncond_sum and noise_pred_text_sum
|
1064 |
-
print(f"noise_pred_uncond_sum min: {noise_pred_uncond_sum.min()} max: {noise_pred_uncond_sum.max()}")
|
1065 |
-
print(f"noise_pred_text_sum min: {noise_pred_text_sum.min()} max: {noise_pred_text_sum.max()}")
|
1066 |
|
1067 |
# set the step index to the current batch
|
1068 |
self.scheduler._step_index = i
|
@@ -1076,7 +1072,6 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
1076 |
|
1077 |
# compute the previous noisy sample x_t -> x_t-1
|
1078 |
latents = self.scheduler.step(noise_pred, t, latents, **extra_step_kwargs).prev_sample
|
1079 |
-
print(f"latents min: {latents.min()} max: {latents.max()}")
|
1080 |
|
1081 |
# call the callback, if provided
|
1082 |
if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
|
|
|
1059 |
noise_pred_uncond_sum[:, :, current_context_start : current_context_start + context_size, :, :] += noise_pred_uncond
|
1060 |
noise_pred_text_sum[:, :, current_context_start : current_context_start + context_size, :, :] += noise_pred_text
|
1061 |
latent_counter[current_context_start : current_context_start + context_size] += 1
|
|
|
|
|
|
|
|
|
1062 |
|
1063 |
# set the step index to the current batch
|
1064 |
self.scheduler._step_index = i
|
|
|
1072 |
|
1073 |
# compute the previous noisy sample x_t -> x_t-1
|
1074 |
latents = self.scheduler.step(noise_pred, t, latents, **extra_step_kwargs).prev_sample
|
|
|
1075 |
|
1076 |
# call the callback, if provided
|
1077 |
if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
|