smoothieAI commited on
Commit
fdaed15
·
verified ·
1 Parent(s): 495c02c

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -1012,7 +1012,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1012
 
1013
  # print step index
1014
  print(f"step index: {self.scheduler.step_index}")
1015
- print(f"sigma: {self.sigmas[self.scheduler.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
 
1012
 
1013
  # print step index
1014
  print(f"step index: {self.scheduler.step_index}")
1015
+ print(f"sigma: {self.scheduler.sigmas[self.scheduler.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