small fix (#12573)

2024-12-18 15:48:22 +08:00 · 2024-12-18 15:48:22 +08:00 · e2ae42929a
commit e2ae42929a
parent a4eb561f36
1 changed files with 1 additions and 1 deletions
--- a/python/llm/src/ipex_llm/transformers/models/minicpmv.py
+++ b/python/llm/src/ipex_llm/transformers/models/minicpmv.py
@ -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,