Unverified Commit 0df83c79 authored by neverix's avatar neverix Committed by GitHub
Browse files

Fixes in comments in SD2 D2I (#1903)

parent 4a7e4cec
...@@ -551,7 +551,7 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline): ...@@ -551,7 +551,7 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline):
prompt, device, num_images_per_prompt, do_classifier_free_guidance, negative_prompt 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( depth_mask = self.prepare_depth_map(
image, image,
depth_map, depth_map,
...@@ -561,10 +561,10 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline): ...@@ -561,10 +561,10 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline):
device, device,
) )
# 5. Prepare depth mask # 5. Preprocess image
image = preprocess(image) image = preprocess(image)
# 6. set timesteps # 6. Set timesteps
self.scheduler.set_timesteps(num_inference_steps, device=device) self.scheduler.set_timesteps(num_inference_steps, device=device)
timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, strength, device) timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, strength, device)
latent_timestep = timesteps[:1].repeat(batch_size * num_images_per_prompt) latent_timestep = timesteps[:1].repeat(batch_size * num_images_per_prompt)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment