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

[NPU] further fix of new_value_states (#12538)

Browse source
This commit is contained in:
Ruonan Wang 2024-12-12 21:42:00 -08:00 committed by GitHub
parent fa261b8af1
commit 7cc01fdc86
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/npu_models/mp_models_base.py
View file

@ -213,6 +213,7 @@ class LLMBaseNNFactory(NNFactory):
value_states = new_value_states
else:
value_states = self.transpose(value_states, [0, 2, 1, 3])
new_value_states = value_states
query_states, key_states = self.apply_rotary_pos_emb(
q=query_states,
@ -225,7 +226,6 @@ class LLMBaseNNFactory(NNFactory):
head_dim=head_dim,
)
new_key_states = key_states
new_value_states = value_states
if mode == "decode":
key_states = self.concat(past_key, key_states, axis=-2)