Daniël de Kok 5b6b74e21d Improve support for GPUs with capability < 8 (#2575) ... * Improve support for GPUs with capability < 8 - For models that cannot use flashinfer, use flash-attn v1 + paged attention for models with a compute capability older than 8. - Disable prefix caching when using paged attention. - When using flash-attn v1, pass the key/value, rather than the cache, since v1 cannot use block tables. * nix: add flash-attn-v1 to the server environment * Move disabling prefix caching into the block of exceptions * Capability as `usize`s		2024-09-27 16:19:42 +02:00
..
__init__.py	feat(server): flash santacoder (#153)	2023-04-03 19:06:42 +02:00
bloom_modeling.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
clip.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
flash_cohere_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_dbrx_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_deepseek_v2_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_gemma2_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_gemma_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_gpt2_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_gptj_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_llama_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_mistral_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_mixtral_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_neox_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_pali_gemma_modeling.py	Lots of improvements (Still 2 allocators) (#2449)	2024-08-29 16:29:01 +02:00
flash_phi_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_qwen2_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_rw_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_santacoder_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
flash_starcoder2_modeling.py	Improve support for GPUs with capability < 8 (#2575)	2024-09-27 16:19:42 +02:00
idefics2.py	Lots of improvements (Still 2 allocators) (#2449)	2024-08-29 16:29:01 +02:00
idefics_config.py	chore: add pre-commit (#1569)	2024-02-16 11:58:58 +01:00
idefics_image_processing.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
idefics_modeling.py	enable HuggingFaceM4/idefics-9b in intel gpu (#2338)	2024-08-01 11:08:36 +02:00
idefics_perceiver.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
idefics_processing.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
idefics_vision.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
llava_next.py	Lots of improvements (Still 2 allocators) (#2449)	2024-08-29 16:29:01 +02:00
mamba_modeling.py	Refactor layers. (#1866)	2024-05-13 12:44:30 +02:00
mpt_modeling.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
neox_modeling.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
opt_modeling.py	Fix the prefix for OPT model in opt_modelling.py #2370 (CI RUN) (#2371)	2024-08-07 23:14:02 -04:00
phi_modeling.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
siglip.py	Fix: don't apply post layernorm in SiglipVisionTransformer (#2459)	2024-08-26 17:04:46 -04:00
t5_modeling.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
vlm.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00