hf_text-generation-inference

Commit Graph

Author	SHA1	Message	Date
fxmarty	406885638b	skip exl2 tests on rocm	2024-06-24 18:49:45 +02: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

406885638b

skip exl2 tests on rocm

2024-06-24 18:49:45 +02: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