huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-07-25 09:20:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
820 Commits 239 Branches 64 Tags 34 MiB
debug-gpt2
Commit Graph

2 Commits

Author SHA1 Message Date
Daniël de Kok
51621439a4 marlin: improve build 2024-06-06 17:19:46 +02:00
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