diffusers
b9d64204 - Fix AuraFlow attn processors applying norm_added_q to key projection (#13533)

Commit
60 days ago
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.
Author
Parents
Loading