text-generation-inference/server/text_generation_server/models
2024-09-25 05:27:40 +00:00
..
custom_modeling Add support for AWQ-quantized Idefics2 (#2233) 2024-09-25 05:27:40 +00:00
__init__.py Falcon/DBRX: get correct number of key-value heads (#2205) 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 Move quantized weight handling out of the Weights class (#2194) 2024-09-25 05:27:40 +00:00
flash_causal_lm.py Move quantized weight handling out of the Weights class (#2194) 2024-09-25 05:27:40 +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 Move quantized weight handling out of the Weights class (#2194) 2024-09-25 05:27:40 +00:00
mamba.py Move quantized weight handling out of the Weights class (#2194) 2024-09-25 05:27:40 +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 Move quantized weight handling out of the Weights class (#2194) 2024-09-25 05:27:40 +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