text-generation-inference/server/text_generation_server
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
..
models feat(server): Add Non flash MPT. (#514) 2023-07-03 13:01:46 +02:00
pb feat(server): clear cache on error (#143) 2023-03-28 11:29:35 +02:00
utils feat(server): Add Non flash MPT. (#514) 2023-07-03 13:01:46 +02:00
__init__.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
cache.py feat(server): add paged attention to flash models (#516) 2023-06-30 19:09:59 +02:00
cli.py feat: Add the option to force another dtype than f16. (#513) 2023-06-30 20:30:09 +02:00
interceptor.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
server.py feat: Add the option to force another dtype than f16. (#513) 2023-06-30 20:30:09 +02:00
tracing.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00