Update pipeline.py
Browse files- 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
|