ipex-llm/python/llm/src/ipex_llm/transformers/models
SONG Ge 192ae35012
Add support for llama2 quantize_kv with transformers 4.38.0 (#11054)
* add support for llama2 quantize_kv with transformers 4.38.0

* fix code style

* fix code style
2024-05-16 22:23:39 +08:00
..
__init__.py Refactor bigdl.llm to ipex_llm (#24) 2024-03-22 15:41:21 +08:00
aquila.py LLM: fix get env KV_CACHE_ALLOC_BLOCK_LENGTH type. (#10771) 2024-04-16 09:32:30 +08:00
baichuan.py use new sdp and fp32 sdp (#11007) 2024-05-14 14:29:18 +08:00
baichuan2.py Use new sdp again (#11025) 2024-05-16 09:33:34 +08:00
bert.py Refactor bigdl.llm to ipex_llm (#24) 2024-03-22 15:41:21 +08:00
bloom.py LLM: fix get env KV_CACHE_ALLOC_BLOCK_LENGTH type. (#10771) 2024-04-16 09:32:30 +08:00
chatglm.py fix chatglm run error (#11045) 2024-05-16 15:39:18 +08:00
chatglm2.py Use new sdp again (#11025) 2024-05-16 09:33:34 +08:00
chatglm2_32k.py LLM: fix get env KV_CACHE_ALLOC_BLOCK_LENGTH type. (#10771) 2024-04-16 09:32:30 +08:00
cohere.py use new sdp and fp32 sdp (#11007) 2024-05-14 14:29:18 +08:00
decilm.py LLM: fix get env KV_CACHE_ALLOC_BLOCK_LENGTH type. (#10771) 2024-04-16 09:32:30 +08:00
falcon.py LLM: fix get env KV_CACHE_ALLOC_BLOCK_LENGTH type. (#10771) 2024-04-16 09:32:30 +08:00
gemma.py remove rms norm copy (#10793) 2024-04-19 13:57:48 +08:00
gptbigcode.py Refactor bigdl.llm to ipex_llm (#24) 2024-03-22 15:41:21 +08:00
gptj.py LLM: fix get env KV_CACHE_ALLOC_BLOCK_LENGTH type. (#10771) 2024-04-16 09:32:30 +08:00
gptneox.py LLM: fix get env KV_CACHE_ALLOC_BLOCK_LENGTH type. (#10771) 2024-04-16 09:32:30 +08:00
internlm.py LLM: fix get env KV_CACHE_ALLOC_BLOCK_LENGTH type. (#10771) 2024-04-16 09:32:30 +08:00
llama.py Add support for llama2 quantize_kv with transformers 4.38.0 (#11054) 2024-05-16 22:23:39 +08:00
mistral.py use new sdp and fp32 sdp (#11007) 2024-05-14 14:29:18 +08:00
mixtral.py use new sdp and fp32 sdp (#11007) 2024-05-14 14:29:18 +08:00
mpt.py LLM: fix get env KV_CACHE_ALLOC_BLOCK_LENGTH type. (#10771) 2024-04-16 09:32:30 +08:00
phi.py remove new_layout parameter (#10906) 2024-04-29 10:31:50 +08:00
phi3.py use new rope in phi3 (#11047) 2024-05-16 15:12:35 +08:00
phixtral.py Disable fast fused rope on UHD (#10780) 2024-04-18 10:03:53 +08:00
qwen.py use new sdp and fp32 sdp (#11007) 2024-05-14 14:29:18 +08:00
qwen2.py use new sdp and fp32 sdp (#11007) 2024-05-14 14:29:18 +08:00
qwen2_moe.py Use new sdp again (#11025) 2024-05-16 09:33:34 +08:00
qwen_vl.py use new sdp and fp32 sdp (#11007) 2024-05-14 14:29:18 +08:00
rwkv4.py fix rwkv with pip installer (#10591) 2024-03-29 17:56:45 +08:00
rwkv5.py disable rwkv5 fp16 (#10699) 2024-04-09 16:42:11 +08:00
stablelm.py use new sdp and fp32 sdp (#11007) 2024-05-14 14:29:18 +08:00
starcoder2.py remove new_layout parameter (#10906) 2024-04-29 10:31:50 +08:00
utils.py Use new sdp again (#11025) 2024-05-16 09:33:34 +08:00
yuan.py use new quantize kv (#10888) 2024-04-26 14:42:17 +08:00