Use t instead of timestep in _apply_perturbed_attention_guidance

hlky · hlky · commit d05bf73f6c5b · 2024-12-16T16:35:32.000Z
diff --git a/src/diffusers/pipelines/pag/pipeline_pag_sana.py b/src/diffusers/pipelines/pag/pipeline_pag_sana.py
@@ -840,7 +840,7 @@ def __call__(
                 # 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)

Original file line number	Diff line number	Diff line change
`@@ -840,7 +840,7 @@ def __call__(`
`840`	`840`	`# perform guidance`
`841`	`841`	`if self.do_perturbed_attention_guidance:`
`842`	`842`	`noise_pred = self._apply_perturbed_attention_guidance(`
`843`		`- noise_pred, self.do_classifier_free_guidance, guidance_scale, timestep`
	`843`	`+ noise_pred, self.do_classifier_free_guidance, guidance_scale, t`
`844`	`844`	`)`
`845`	`845`	`elif self.do_classifier_free_guidance:`
`846`	`846`	`noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)`