hf_text-generation-inference/server/text_generation_server
Daniël de Kok 4594e6faba Add support for Marlin-quantized models
This change adds support for Marlin-quantized models. Marlin is an
FP16xINT4 matmul kernel, which provides good speedups decoding batches
of 16-32 tokens. It supports quantized models with symmetric
quantization, groupsize -1 or 128, and 4-bit.

Tested with:

- Llama 2
- Llama 3
- Phi 3
2024-06-06 13:16:52 +02:00
..
layers Add support for Marlin-quantized models 2024-06-06 13:16:52 +02:00
models Add support for Marlin-quantized models 2024-06-06 13:16:52 +02:00
pb chore: add pre-commit (#1569) 2024-02-16 11:58:58 +01:00
utils Add support for Marlin-quantized models 2024-06-06 13:16:52 +02:00
__init__.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00
cache.py fix(server): decrease memory fragmentation (#557) 2023-07-06 14:28:33 +02:00
cli.py Add support for Marlin-quantized models 2024-06-06 13:16:52 +02:00
interceptor.py v2.0.0 (#1736) 2024-04-12 18:38:34 +02:00
server.py Add support for exl2 quantization 2024-05-30 11:28:05 +02:00
tracing.py feat(clients): Python client (#103) 2023-03-07 18:52:22 +01:00