AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

Fixes in comments in SD2 D2I (#1903)

Browse Source
This commit is contained in:
neverix
2023-01-03 19:24:36 +04:00
committed by GitHub
parent 4a7e4cec38
commit 0df83c79e4
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -551,7 +551,7 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline):
prompt, device, num_images_per_prompt, do_classifier_free_guidance, negative_prompt
)
# 4. Preprocess image
# 4. Prepare depth mask
depth_mask = self.prepare_depth_map(
image,
depth_map,
@@ -561,10 +561,10 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline):
device,
)
# 5. Prepare depth mask
# 5. Preprocess image
image = preprocess(image)
# 6. set timesteps
# 6. Set timesteps
self.scheduler.set_timesteps(num_inference_steps, device=device)
timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, strength, device)
latent_timestep = timesteps[:1].repeat(batch_size * num_images_per_prompt)