Skip to content

Commit d74483c

Browse files
authoredNov 14, 2024
Fix Progress Bar Updates in SD 1.5 PAG Img2Img pipeline (#9925)
fix progress bar updates in SD 1.5 PAG Img2Img pipeline
1 parent 1dbd26f commit d74483c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed
 

‎src/diffusers/pipelines/pag/pipeline_pag_sd_img2img.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1063,6 +1063,9 @@ def __call__(
10631063
prompt_embeds = callback_outputs.pop("prompt_embeds", prompt_embeds)
10641064
negative_prompt_embeds = callback_outputs.pop("negative_prompt_embeds", negative_prompt_embeds)
10651065

1066+
if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
1067+
progress_bar.update()
1068+
10661069
if not output_type == "latent":
10671070
image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False, generator=generator)[
10681071
0

0 commit comments

Comments
 (0)