AlanB commited on
Commit
5cbad96
1 Parent(s): 0370e41

Add num_timesteps for callback kwargs

Browse files
Files changed (1) hide show
  1. 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