hf_text-generation-inference/server/text_generation_server/utils
Nicolas Patry 1da07e85aa
feat(server): Add Non flash MPT. (#514)
# What does this PR do?


This adds a non flash version of MPT.
Flash is harder because we need to create a bias ready cuda kernel of
flash attention.

Fixes
https://github.com/huggingface/text-generation-inference/issues/361
Fixes
https://github.com/huggingface/text-generation-inference/issues/491
Fixes
https://github.com/huggingface/text-generation-inference/issues/290
2023-07-03 13:01:46 +02:00
..
gptq feat(server): Add inference support for GPTQ (llama + falcon tested) + Quantization script (#438) 2023-06-26 12:27:01 +02:00
__init__.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
convert.py feat(server): Update convert logic. (#483) 2023-06-23 12:40:46 +02:00
dist.py fix(server): Do not init process group if already initialized (#388) 2023-06-26 12:32:54 +02:00
hub.py feat(server): Adding new ignore_rule for conversion. (#485) 2023-06-23 12:41:13 +02:00
layers.py feat(server): Add Non flash MPT. (#514) 2023-07-03 13:01:46 +02:00
logits_process.py fix(server): fix warpers on CPU (#472) 2023-06-20 11:06:10 +02:00
tokens.py feat(server): add paged attention to flash models (#516) 2023-06-30 19:09:59 +02:00
watermark.py fix(server): fix flash-neox scores warping (#137) 2023-03-24 18:21:41 +01:00
weights.py feat(server): add paged attention to flash models (#516) 2023-06-30 19:09:59 +02:00