text-generation-inference/server/text_generation_server/utils
2023-07-19 17:55:41 +00:00
..
gptq fix per-column quantization 2023-07-19 17:55:41 +00:00
__init__.py feat(server): Rework model loading (#344) 2023-06-08 14:51:52 +02:00
convert.py fix(server): blacklist local files (#609) 2023-07-13 21:54:55 +02:00
dist.py fix(server): Do not init process group if already initialized (#388) 2023-06-26 12:32:54 +02:00
flash_attn.py feat(server): flash attention v2 (#624) 2023-07-18 16:21:18 +02:00
hub.py feat(server): Adding new ignore_rule for conversion. (#485) 2023-06-23 12:41:13 +02:00
layers.py Merge branch 'main' into gptq-cuda-kernels 2023-07-12 18:47:30 +02:00
logits_process.py fix(server): avoid errors for very small top_p values (#544) 2023-07-04 20:11:33 +02:00
tokens.py feat(server): add paged attention to flash models (#516) 2023-06-30 19:09:59 +02:00
watermark.py fix(server): fix flash-neox scores warping (#137) 2023-03-24 18:21:41 +01:00
weights.py fix per-column quantization 2023-07-19 17:55:41 +00:00