Skip to content

Commit 0df83c7

Browse files
authored
Fixes in comments in SD2 D2I (huggingface#1903)
1 parent 4a7e4ce commit 0df83c7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -551,7 +551,7 @@ def __call__(
551551
prompt, device, num_images_per_prompt, do_classifier_free_guidance, negative_prompt
552552
)
553553

554-
# 4. Preprocess image
554+
# 4. Prepare depth mask
555555
depth_mask = self.prepare_depth_map(
556556
image,
557557
depth_map,
@@ -561,10 +561,10 @@ def __call__(
561561
device,
562562
)
563563

564-
# 5. Prepare depth mask
564+
# 5. Preprocess image
565565
image = preprocess(image)
566566

567-
# 6. set timesteps
567+
# 6. Set timesteps
568568
self.scheduler.set_timesteps(num_inference_steps, device=device)
569569
timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, strength, device)
570570
latent_timestep = timesteps[:1].repeat(batch_size * num_images_per_prompt)

0 commit comments

Comments
 (0)