Daniël de Kok 5726a9ca81 Move to moe-kernels package and switch to common MoE layer ... This change introduces the new `moe-kernels` package: - Add `moe-kernels` as a dependency. - Introduce a `SparseMoELayer` module that can be used by MoE models. - Port over Mixtral and Deepseek.		2024-09-16 10:57:44 +00: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	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_dbrx_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_deepseek_v2_modeling.py	Move to moe-kernels package and switch to common MoE layer	2024-09-16 10:57:44 +00:00
flash_gemma2_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_gemma_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_gpt2_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_gptj_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_llama_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_mistral_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_mixtral_modeling.py	Move to moe-kernels package and switch to common MoE layer	2024-09-16 10:57:44 +00:00
flash_neox_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +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	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_qwen2_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_rw_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_santacoder_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +02:00
flash_starcoder2_modeling.py	hotfix: fix regression of attention api change in intel platform (#2439)	2024-09-05 17:41:39 +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