AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

Update src/diffusers/models/transformers/transformer_wan.py

Browse Source
This commit is contained in:
Aryan
2025-04-05 03:41:25 +05:30
committed by GitHub
parent 74e34e5f69
commit 1411b33563
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/transformers/transformer_wan.py
View File

@@ -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)