AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-10-27 21:33:02 +03:00
Code Activity

up

Browse Source
This commit is contained in:
Patrick von Platen
2022-12-10 17:28:02 +00:00
parent 1c0b63c0f6
commit 2b76136885
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py
View File

@@ -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

2
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py
View File

@@ -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

2
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
View File

@@ -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