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

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -1011,8 +1011,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
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
 
1011
  print(f"sigma len: {len(self.scheduler.sigmas)}")
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