This commit is contained in:
Yishuo Wang 2024-12-18 15:48:22 +08:00 committed by GitHub
parent a4eb561f36
commit e2ae42929a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -62,7 +62,7 @@ def siglip_attention_forward(
import xe_addons
attn_weights = None
attn_output = xe_addons.siglip_sdp_non_causal(query_states, key_states,
value_states, attention_softmax)
value_states, attention_mask)
else:
query_states, key_states, value_states = padding_qkv_hd(
query_states, key_states, value_states,