hf_text-generation-inference/server/text_generation_server/models/custom_modeling
OlivierDehaene f04255c694
feat: Add dbrx support (#1685)
Close #1679
2024-03-29 18:49:36 +01:00
..
__init__.py feat(server): flash santacoder (#153) 2023-04-03 19:06:42 +02:00
bloom_modeling.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
flash_cohere_modeling.py feat: cohere (#1660) 2024-03-22 17:59:25 +01:00
flash_dbrx_modeling.py feat: Add dbrx support (#1685) 2024-03-29 18:49:36 +01:00
flash_gemma_modeling.py feat: cohere (#1660) 2024-03-22 17:59:25 +01:00
flash_llama_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_mistral_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_mixtral_modeling.py feat: Add dbrx support (#1685) 2024-03-29 18:49:36 +01:00
flash_neox_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_phi_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_qwen2_modeling.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
flash_rw_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_santacoder_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
flash_starcoder2_modeling.py feat: starcoder2 (#1605) 2024-02-28 12:07:08 +01:00
idefics_config.py chore: add pre-commit (#1569) 2024-02-16 11:58:58 +01:00
idefics_image_processing.py chore: formatting 2023-12-11 14:49:52 +01:00
idefics_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
idefics_perceiver.py feat: format code (#1070) 2023-09-27 12:22:09 +02:00
idefics_processing.py chore: add pre-commit (#1569) 2024-02-16 11:58:58 +01:00
idefics_vision.py feat: format code (#1070) 2023-09-27 12:22:09 +02:00
mamba_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
mpt_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
neox_modeling.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
opt_modeling.py feat: Qwen2 (#1608) 2024-02-28 15:50:31 +01:00
phi_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00
t5_modeling.py Revamp medusa implementation so that every model can benefit. (#1588) 2024-02-26 19:49:28 +01:00