From 1411b33563899125bb4600874d49dbb99e18eb9e Mon Sep 17 00:00:00 2001 From: Aryan Date: Sat, 5 Apr 2025 03:41:25 +0530 Subject: [PATCH] Update src/diffusers/models/transformers/transformer_wan.py --- src/diffusers/models/transformers/transformer_wan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/transformers/transformer_wan.py b/src/diffusers/models/transformers/transformer_wan.py index 703e2f82c2..cf9decfc9e 100644 --- a/src/diffusers/models/transformers/transformer_wan.py +++ b/src/diffusers/models/transformers/transformer_wan.py @@ -537,7 +537,7 @@ class WanPAGAttnProcessor2_0: query, key, value, attn_mask=attention_mask, dropout_p=0.0, is_causal=False ) else: - # Perturbed attention applied only when self-attention + # Perturbed attention applied only to self-attention path hidden_states = value hidden_states = hidden_states.transpose(1, 2).flatten(2, 3)