text-generation-inference/server/text_generation_server/utils
2024-04-29 14:55:59 +00:00
..
awq ROCm AWQ support (#1514) 2024-02-09 10:45:16 +01:00
gptq chore: add pre-commit (#1569) 2024-02-16 11:58:58 +01: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 feat: add cuda memory fraction (#659) 2023-07-24 11:43:58 +02:00
flash_attn_triton.py add missing files 2024-04-19 13:46:54 +00:00
flash_attn.py black 2024-04-23 15:05:39 +00:00
hub.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
import_utils.py Add RoCm support (#1243) 2023-11-27 14:08:12 +01:00
layers.py Merge branch 'main' into mi300-compat 2024-04-26 11:28:42 +02:00
log.py v1.3.4 2023-12-22 15:46:04 +01:00
logits_process.py fix: avoid frequency and repetition penalty on padding tokens (#1765) 2024-04-23 23:19:16 +02:00
paged_attention.py working & cached tunableop 2024-04-29 14:55:59 +00: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 Phi3 support (#1797) 2024-04-23 18:40:05 +02:00