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

fix typoes

Browse source
This commit is contained in:
Huang, Xinshengzi 2024-08-22 11:09:58 +08:00
parent 6bb9035788
commit 6a5ca17afc
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/baichuan.py
View file

@ -283,7 +283,7 @@ def baichuan_attention_forward_7b(
key_states, value_states, self.layer_idx, key_states, value_states, self.layer_idx,
query_states, attention_mask, 1, query_states, attention_mask, 1,
self.config, enough_kv_room, KV_CACHE_ALLOC_BLOCK_LENGTH) self.config, enough_kv_room, KV_CACHE_ALLOC_BLOCK_LENGTH)
if self.training: if self.training:
warnings.warn("xops is not supported on Intel GPU, so just use normal implementation") warnings.warn("xops is not supported on Intel GPU, so just use normal implementation")