text-generation-inference/server/text_generation_server/models/custom_modeling
OlivierDehaene 0c207f71ed feat: experimental support for cuda graphs (#1428)
Co-authored-by: Nicolas Patry <patry.nicolas@protonmail.com>
2024-04-24 13:15:45 +03:00
..
__init__.py feat(server): flash santacoder (#153) 2023-04-03 19:06:42 +02:00
bloom_modeling.py feat: format code (#1070) 2023-09-27 12:22:09 +02:00
flash_llama_modeling.py chore: formatting 2024-04-18 16:26:00 +03:00
flash_mistral_modeling.py feat: experimental support for cuda graphs (#1428) 2024-04-24 13:15:45 +03:00
flash_mixtral_modeling.py feat: experimental support for cuda graphs (#1428) 2024-04-24 13:15:45 +03:00
flash_neox_modeling.py GPTNeoX: Use static rotary embedding (#1498) 2024-04-23 09:21:21 +03:00
flash_phi_modeling.py v1.4.0 (#1494) 2024-04-22 15:47:42 +03:00
flash_rw_modeling.py Add RoCm support (#1243) 2023-11-27 14:08:12 +01:00
flash_santacoder_modeling.py ROCm AWQ support (#1514) 2024-04-24 09:21:34 +00:00
idefics_config.py small fix on idefics (#954) 2023-09-01 18:44:34 +02:00
idefics_image_processing.py chore: formatting 2024-04-18 16:26:00 +03:00
idefics_modeling.py chore: formatting 2024-04-18 16:26:00 +03:00
idefics_perceiver.py feat: format code (#1070) 2023-09-27 12:22:09 +02:00
idefics_processing.py chore: formatting 2024-04-18 16:26:00 +03:00
idefics_vision.py feat: format code (#1070) 2023-09-27 12:22:09 +02:00
mamba_modeling.py feat(server): add frequency penalty (#1541) 2024-04-24 08:43:50 +00:00
mpt_modeling.py v1.4.0 (#1494) 2024-04-22 15:47:42 +03:00
neox_modeling.py Fix: Replace view() with reshape() in neox_modeling.py to resolve RuntimeError (#1155) 2023-10-19 11:54:26 +02:00
opt_modeling.py Complete FastLinear.load parameters in OPTDecoder initialization (#1060) 2023-09-27 12:25:59 +02:00
phi_modeling.py v1.4.0 (#1494) 2024-04-22 15:47:42 +03:00
t5_modeling.py Fixing t5 loading. (#1042) 2023-09-25 12:22:28 +02:00