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

painebenjamin · rootonchair · web-flow · commit d49cbc9963a5 · 2024-11-20T17:15:12.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
@@ -958,9 +958,6 @@ def __call__(
                 generator,
             )
 
-        num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)
-        self._num_timesteps = len(timesteps)
-
         if self.do_perturbed_attention_guidance:
             original_attn_proc = self.transformer.attn_processors
             self._set_pag_attn_processor(

Original file line number	Diff line number	Diff line change
`@@ -958,9 +958,6 @@ def __call__(`
`958`	`958`	`generator,`
`959`	`959`	`)`
`960`	`960`
`961`		`- num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)`
`962`		`- self._num_timesteps = len(timesteps)`
`963`		`-`
`964`	`961`	`if self.do_perturbed_attention_guidance:`
`965`	`962`	`original_attn_proc = self.transformer.attn_processors`
`966`	`963`	`self._set_pag_attn_processor(`