hf_text-generation-inference/server/text_generation_server/models
Daniël de Kok 5c7c9f1390
Falcon/DBRX: get correct number of key-value heads (#2205)
2024-07-08 13:22:38 +02:00
..
custom_modeling Falcon/DBRX: get correct number of key-value heads (#2205) 2024-07-08 13:22:38 +02:00
__init__.py Falcon/DBRX: get correct number of key-value heads (#2205) 2024-07-08 13:22:38 +02:00
bloom.py Refactor dead code - Removing all `flash_xxx.py` files. (#2166) 2024-07-05 10:29:56 +02:00
causal_lm.py Consistently take `prefix` in model constructors (#2191) 2024-07-05 16:07:48 +02:00
flash_causal_lm.py Falcon/DBRX: get correct number of key-value heads (#2205) 2024-07-08 13:22:38 +02:00
flash_mistral.py Refactor dead code - Removing all `flash_xxx.py` files. (#2166) 2024-07-05 10:29:56 +02:00
galactica.py Refactor dead code - Removing all `flash_xxx.py` files. (#2166) 2024-07-05 10:29:56 +02:00
globals.py [Major Change][Undecided yet] Move to FlashDecoding instead of PagedAttention kernel. (#1940) 2024-07-01 23:28:00 +02:00
idefics.py Enable multiple LoRa adapters (#2010) 2024-06-25 14:46:27 -04:00
idefics_causal_lm.py Enable multiple LoRa adapters (#2010) 2024-06-25 14:46:27 -04:00
mamba.py Enable multiple LoRa adapters (#2010) 2024-06-25 14:46:27 -04:00
model.py Hotfixing after refactor. 2024-07-05 09:25:29 +00:00
pali_gemma.py Refactor dead code - Removing all `flash_xxx.py` files. (#2166) 2024-07-05 10:29:56 +02:00
seq2seq_lm.py Refactor dead code - Removing all `flash_xxx.py` files. (#2166) 2024-07-05 10:29:56 +02:00
types.py chore: add pre-commit (#1569) 2024-02-16 11:58:58 +01:00
vlm_causal_lm.py Refactor dead code - Removing all `flash_xxx.py` files. (#2166) 2024-07-05 10:29:56 +02:00