text-generation-inference

huggingface/text-generation-inference

Fork 0

mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-24 00:12:08 +00:00

Commit Graph

Author	SHA1	Message	Date
fxmarty	73b067d193	skip exl2 tests on rocm	2024-06-11 09:29:08 +00:00
Danië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

Author

SHA1

Message

Date

fxmarty

73b067d193

skip exl2 tests on rocm

2024-06-11 09:29:08 +00:00

Danië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

2 Commits