text-generation-inference/server/text_generation_server/utils
2024-04-19 14:18:05 +03:00
..
awq/quantize feat: format code (#1070) 2023-09-27 12:22:09 +02:00
gptq fix: fix quant linear autotune 2024-04-19 14:17:39 +03:00
__init__.py Add Habana copyright header (#122) 2024-04-08 18:06:21 +02:00
convert.py fit for baichuan models (#981) 2023-09-08 16:51:34 +02:00
debug.py Add Habana copyright header (#122) 2024-04-08 18:06:21 +02:00
dist.py Add changes from Optimum Habana's TGI folder 2023-12-05 11:12:16 +01:00
flash_attn.py fix: max_past default value must be -1, not 0 (#1348) 2024-04-19 14:18:05 +03: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 feat: add quant to mixtral (#1337) 2024-04-18 16:32:50 +03:00
logits_process.py Add changes from Optimum Habana's TGI folder 2023-12-05 11:12:16 +01:00
medusa.py chore: formatting 2024-04-18 16:26:00 +03:00
paged_attention.py chore: formatting 2024-04-18 16:26:00 +03:00
peft.py chore: formatting 2024-04-18 16:26:00 +03:00
speculate.py chore: formatting 2024-04-18 16:26:00 +03:00
tokens.py fix: only keep stop sequence buffer if we have some 2024-04-19 14:18:00 +03:00
watermark.py Add changes from Optimum Habana's TGI folder 2023-12-05 11:12:16 +01:00
weights.py fix: fix gpt-q params loading 2024-04-19 12:12:50 +03:00