diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py b/src/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py index 0ceb2eb3d8..29b148a918 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py @@ -564,7 +564,7 @@ class CycleDiffusionPipeline(DiffusionPipeline): # 5. Prepare timesteps self.scheduler.set_timesteps(num_inference_steps, device=device) - timesteps, num_inference_steps, latent_timestep = self.get_timesteps(num_inference_steps, strength, device) + timesteps, latent_timestep, num_inference_steps = self.get_timesteps(num_inference_steps, strength, device) latent_timestep = latent_timestep.repeat(batch_size * num_images_per_prompt) # 6. Prepare latent variables diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py index 66432f1668..78040b1bae 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py @@ -520,7 +520,7 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline): # 6. set timesteps self.scheduler.set_timesteps(num_inference_steps, device=device) - timesteps, num_inference_steps, latent_timestep = self.get_timesteps(num_inference_steps, strength, device) + timesteps, latent_timestep, num_inference_steps = self.get_timesteps(num_inference_steps, strength, device) latent_timestep = latent_timestep.repeat(batch_size * num_images_per_prompt) # 7. Prepare latent variables diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py index ec360a3cca..f08c15227e 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py @@ -534,7 +534,7 @@ class StableDiffusionInpaintPipelineLegacy(DiffusionPipeline): # 5. set timesteps self.scheduler.set_timesteps(num_inference_steps, device=device) - timesteps, num_inference_steps, latent_timestep = self.get_timesteps(num_inference_steps, strength, device) + timesteps, latent_timestep, num_inference_steps = self.get_timesteps(num_inference_steps, strength, device) latent_timestep = latent_timestep.repeat(batch_size * num_images_per_prompt) # 6. Prepare latent variables