Use `t` instead of `timestep` in `_apply_perturbed_attention_guidance` (#10243)

672bd495 · hlky · GitHub · ea893a9a · 672bd495
Unverified Commit 672bd495 authored Dec 16, 2024 by hlky Committed by GitHub Dec 16, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/pipelines/pag/pipeline_pag_sana.py src/diffusers/pipelines/pag/pipeline_pag_sana.py +1 -1

No files found.
--- a/src/diffusers/pipelines/pag/pipeline_pag_sana.py
+++ b/src/diffusers/pipelines/pag/pipeline_pag_sana.py
@@ -840,7 +840,7 @@ class SanaPAGPipeline(DiffusionPipeline, PAGMixin):
                # perform guidance
                if self.do_perturbed_attention_guidance:
                    noise_pred = self._apply_perturbed_attention_guidance(
-                        noise_pred, self.do_classifier_free_guidance, guidance_scale, timestep
+                        noise_pred, self.do_classifier_free_guidance, guidance_scale, t
                    )
                elif self.do_classifier_free_guidance:
                    noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)