text-generation-inference/server/text_generation_server/layers/gptq
Daniël de Kok 36dd16017c Add support for exl2 quantization
Mostly straightforward, changes to existing code:

* Wrap quantizer parameters in a small wrapper to avoid passing
  around untyped tuples and needing to repack them as a dict.
* Move scratch space computation to warmup, because we need the
  maximum input sequence length to avoid allocating huge
  scratch buffers that OOM.
2024-05-30 11:28:05 +02:00
..
__init__.py Add support for exl2 quantization 2024-05-30 11:28:05 +02:00
custom_autotune.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
exllama.py Add support for exl2 quantization 2024-05-30 11:28:05 +02:00
exllamav2.py Add support for exl2 quantization 2024-05-30 11:28:05 +02:00
quant_linear.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00
quantize.py Refactor layers. (#1866) 2024-05-13 12:44:30 +02:00