huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-22 15:32:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
941d36f3fd
text-generation-inference/server/text_generation_server/utils
History
jkaniecki a490847702
Sequence bucketing for prefill (#39) (#67) 
Co-authored-by: mswiniarsk <156412439+mswiniarsk@users.noreply.github.com>
2024-02-23 01:52:14 +01:00
..
awq/quantize feat: format code (#1070) 2023-09-27 12:22:09 +02:00
gptq Exllama v2 (#1211) 2023-11-25 22:38:38 +01:00
__init__.py Make tokenizer optional (#12) 2024-01-19 15:12:04 +01:00
convert.py fit for baichuan models (#981) 2023-09-08 16:51:34 +02:00
dist.py Add changes from Optimum Habana's TGI folder 2023-12-05 11:12:16 +01:00
flash_attn.py Add RoCm support (#1243) 2023-11-27 14:08:12 +01:00
hub.py feat(server): Adding new ignore_rule for conversion. (#485) 2023-06-23 12:41:13 +02:00
import_utils.py Add RoCm support (#1243) 2023-11-27 14:08:12 +01:00
layers.py Add RoCm support (#1243) 2023-11-27 14:08:12 +01:00
logits_process.py Add changes from Optimum Habana's TGI folder 2023-12-05 11:12:16 +01:00
paged_attention.py feat: paged attention v2 (#1183) 2023-10-23 12:29:25 +02:00
peft.py Fix missing trust_remote_code flag for AutoTokenizer in utils.peft (#1270) 2023-11-23 12:41:05 +01:00
tokens.py Sequence bucketing for prefill (#39) (#67) 2024-02-23 01:52:14 +01:00
watermark.py Add changes from Optimum Habana's TGI folder 2023-12-05 11:12:16 +01:00
weights.py Exllama v2 (#1211) 2023-11-25 22:38:38 +01:00