smoothieAI commited on
Commit
681bb9b
·
verified ·
1 Parent(s): a467841

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +4 -0
pipeline.py CHANGED
@@ -1010,6 +1010,10 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1010
  # print size of sigmas
1011
  print(f"sigma len: {len(self.scheduler.sigmas)}")
1012
 
 
 
 
 
1013
  # compute the previous noisy sample x_t -> x_t-1
1014
  current_context_latents = self.scheduler.step(noise_pred, t, current_context_latents, **extra_step_kwargs).prev_sample
1015
 
 
1010
  # print size of sigmas
1011
  print(f"sigma len: {len(self.scheduler.sigmas)}")
1012
 
1013
+ # print step index
1014
+ print(f"step index: {self.step_index}")
1015
+ print(f"sigma: {self.sigmas[self.step_index + 1]}")
1016
+
1017
  # compute the previous noisy sample x_t -> x_t-1
1018
  current_context_latents = self.scheduler.step(noise_pred, t, current_context_latents, **extra_step_kwargs).prev_sample
1019