smoothieAI commited on
Commit
24453ad
·
verified ·
1 Parent(s): c309ef5

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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):