From 2b76136885ab1538fcbb47e7100313cc2abd0179 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Sat, 10 Dec 2022 17:28:02 +0000 Subject: [PATCH] up --- .../pipelines/stable_diffusion/pipeline_cycle_diffusion.py | 2 +- .../stable_diffusion/pipeline_stable_diffusion_depth2img.py | 2 +- .../pipeline_stable_diffusion_inpaint_legacy.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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