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	feat(server): Add native support for PEFT Lora models (#762)	2023-08-03 17:22:45 +02:00
convert.py	Force weights_only (before fully breaking pickle files anyway). (#1710)	2024-04-05 19:23:57 +02:00
dist.py	add intel xpu support for TGI (#1475)	2024-04-26 15:48:58 +02:00
flash_attn_triton.py	MI300 compatibility (#1764)	2024-05-17 15:30:47 +02:00
flash_attn.py	reenable xpu for tgi (#1939)	2024-05-23 14:11:08 +02:00
hub.py	Fixing the download strategy for ibm-fms (#1917)	2024-05-18 13:31:24 +02:00
import_utils.py	reenable xpu for tgi (#1939)	2024-05-23 14:11:08 +02:00
log.py	v1.3.4	2023-12-22 15:46:04 +01:00
logits_process.py	Fixing frequency penalty (#1811)	2024-04-30 12:13:23 +02:00
paged_attention.py	MI300 compatibility (#1764)	2024-05-17 15:30:47 +02:00
peft.py	fix: fix local loading for .bin models (#1419)	2024-01-09 15:21:00 +01:00
speculate.py	chore: formatting	2023-12-11 14:49:52 +01:00
tokens.py	Use the generation config. (#1808)	2024-04-25 19:41:50 +02:00
watermark.py	Fixing watermark. (#851)	2023-08-16 07:17:26 +02:00
weights.py	Add support for exl2 quantization	2024-05-30 11:28:05 +02:00