Skip to content

Commit 3b07354

Browse files
committed
Update pipeline_pag_controlnet_sd_img2img.py
1 parent c858e11 commit 3b07354

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/pipelines/pag/pipeline_pag_controlnet_sd_img2img.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1155,6 +1155,7 @@ def __call__(
11551155
)
11561156
elif self.do_classifier_free_guidance:
11571157
prompt_embeds = torch.cat([negative_prompt_embeds, prompt_embeds])
1158+
11581159
# 3.2 Encode ip_adapter_image
11591160
if ip_adapter_image is not None or ip_adapter_image_embeds is not None:
11601161
ip_adapter_image_embeds = self.prepare_ip_adapter_image_embeds(
@@ -1329,7 +1330,7 @@ def __call__(
13291330
)
13301331
elif self.do_classifier_free_guidance:
13311332
noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
1332-
noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)
1333+
noise_pred = noise_pred_uncond + self.guidance_scale * (noise_pred_text - noise_pred_uncond)
13331334

13341335
# compute the previous noisy sample x_t -> x_t-1
13351336
latents = self.scheduler.step(noise_pred, t, latents, **extra_step_kwargs, return_dict=False)[0]

0 commit comments

Comments
 (0)