Add num_timesteps for callback kwargs
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
@@ -897,6 +897,7 @@ class FluxDifferentialImg2ImgPipeline(DiffusionPipeline, FluxLoraLoaderMixin):
|
|
897 |
# end diff diff preparation
|
898 |
|
899 |
num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)
|
|
|
900 |
|
901 |
# 6. Denoising loop
|
902 |
latents_dtype = latents.dtype
|
|
|
897 |
# end diff diff preparation
|
898 |
|
899 |
num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)
|
900 |
+
self._num_timesteps = len(timesteps)
|
901 |
|
902 |
# 6. Denoising loop
|
903 |
latents_dtype = latents.dtype
|