Unverified Commit d74483c4 authored by Benjamin Paine's avatar Benjamin Paine Committed by GitHub
Browse files

Fix Progress Bar Updates in SD 1.5 PAG Img2Img pipeline (#9925)

fix progress bar updates in SD 1.5 PAG Img2Img pipeline
parent 1dbd26fa
......@@ -1063,6 +1063,9 @@ class StableDiffusionPAGImg2ImgPipeline(
prompt_embeds = callback_outputs.pop("prompt_embeds", prompt_embeds)
negative_prompt_embeds = callback_outputs.pop("negative_prompt_embeds", negative_prompt_embeds)
if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
progress_bar.update()
if not output_type == "latent":
image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False, generator=generator)[
0
......
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