ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

small fix (#12573)

Browse source
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
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/llm/src/ipex_llm/transformers/models/minicpmv.py
View file

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