text-generation-inference/server/text_generation_server/models
Daniël de Kok f11fd699b6 hotfix: Fix number of KV heads (#2202)
Fix number of KV heads
2024-09-25 05:21:34 +00:00
..
custom_modeling fix dbrx & opt model prefix bug (#2201) 2024-09-25 05:21:34 +00:00
__init__.py Consistently take prefix in model constructors (#2191) 2024-09-25 05:21:34 +00:00
bloom.py Refactor dead code - Removing all flash_xxx.py files. (#2166) 2024-09-25 05:20:28 +00:00
causal_lm.py Consistently take prefix in model constructors (#2191) 2024-09-25 05:21:34 +00:00
flash_causal_lm.py hotfix: Fix number of KV heads (#2202) 2024-09-25 05:21:34 +00:00
flash_mistral.py Refactor dead code - Removing all flash_xxx.py files. (#2166) 2024-09-25 05:20:28 +00:00
galactica.py Refactor dead code - Removing all flash_xxx.py files. (#2166) 2024-09-25 05:20:28 +00:00
globals.py Move to FlashDecoding instead of PagedAttention kernel. (#1940) 2024-09-24 03:58:13 +00:00
idefics_causal_lm.py Enable multiple LoRa adapters (#2010) 2024-09-24 03:55:04 +00:00
idefics.py Enable multiple LoRa adapters (#2010) 2024-09-24 03:55:04 +00:00
mamba.py Enable multiple LoRa adapters (#2010) 2024-09-24 03:55:04 +00:00
model.py Hotfixing after refactor. 2024-09-25 05:20:28 +00:00
pali_gemma.py Refactor dead code - Removing all flash_xxx.py files. (#2166) 2024-09-25 05:20:28 +00:00
seq2seq_lm.py Refactor dead code - Removing all flash_xxx.py files. (#2166) 2024-09-25 05:20:28 +00:00
types.py chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
vlm_causal_lm.py Refactor dead code - Removing all flash_xxx.py files. (#2166) 2024-09-25 05:20:28 +00:00