Update src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py

painebenjamin · rootonchair · web-flow · commit 6ed9607f5f53 · 2024-11-20T17:15:06.000-05:00
Co-authored-by: Vinh H. Pham  &lt;phamvinh257@gmail.com&gt;
diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py b/src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py
@@ -969,6 +969,8 @@ def __call__(
             )
 
         # 6. Denoising loop
+        num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)
+        self._num_timesteps = len(timesteps)
         with self.progress_bar(total=num_inference_steps) as progress_bar:
             for i, t in enumerate(timesteps):
                 if self.interrupt:

Original file line number	Diff line number	Diff line change
`@@ -969,6 +969,8 @@ def __call__(`
`969`	`969`	`)`
`970`	`970`
`971`	`971`	`# 6. Denoising loop`
	`972`	`+ num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)`
	`973`	`+ self._num_timesteps = len(timesteps)`
`972`	`974`	`with self.progress_bar(total=num_inference_steps) as progress_bar:`
`973`	`975`	`for i, t in enumerate(timesteps):`
`974`	`976`	`if self.interrupt:`