From b9d6420447113008cef191faa6fcabb01acb1b8b Mon Sep 17 00:00:00 2001 From: Ricardo-M-L <69202550+Ricardo-M-L@users.noreply.github.com> Date: Wed, 22 Apr 2026 03:20:54 +0800 Subject: [PATCH] Fix AuraFlow attn processors applying norm_added_q to key projection (#13533) Both AuraFlowAttnProcessor2_0 and FusedAuraFlowAttnProcessor2_0 were calling attn.norm_added_q on encoder_hidden_states_key_proj while guarded by a check on attn.norm_added_k. This applies the query normalization layer to the key, which is a copy-paste error. Consistent with every other attention processor in this file that defines both norm_added_q and norm_added_k (e.g. FluxAttnProcessor, CogVideoXAttnProcessor, HunyuanAttnProcessor), where norm_added_k is applied to the added key projection. --- src/diffusers/models/attention_processor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/models/attention_processor.py b/src/diffusers/models/attention_processor.py index f83a1753da..e2ece5cb36 100755 --- a/src/diffusers/models/attention_processor.py +++ b/src/diffusers/models/attention_processor.py @@ -2140,7 +2140,7 @@ class AuraFlowAttnProcessor2_0: if attn.norm_added_q is not None: encoder_hidden_states_query_proj = attn.norm_added_q(encoder_hidden_states_query_proj) if attn.norm_added_k is not None: - encoder_hidden_states_key_proj = attn.norm_added_q(encoder_hidden_states_key_proj) + encoder_hidden_states_key_proj = attn.norm_added_k(encoder_hidden_states_key_proj) query = torch.cat([encoder_hidden_states_query_proj, query], dim=1) key = torch.cat([encoder_hidden_states_key_proj, key], dim=1) @@ -2237,7 +2237,7 @@ class FusedAuraFlowAttnProcessor2_0: if attn.norm_added_q is not None: encoder_hidden_states_query_proj = attn.norm_added_q(encoder_hidden_states_query_proj) if attn.norm_added_k is not None: - encoder_hidden_states_key_proj = attn.norm_added_q(encoder_hidden_states_key_proj) + encoder_hidden_states_key_proj = attn.norm_added_k(encoder_hidden_states_key_proj) query = torch.cat([encoder_hidden_states_query_proj, query], dim=1) key = torch.cat([encoder_hidden_states_key_proj, key], dim=1)