huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-07-27 18:30:16 +00:00
Code Issues Packages Projects Releases Wiki Activity
feature/moe-kernels
text-generation-inference/integration-tests/models/__snapshots__/test_flash_mixtral
History
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
..
test_flash_mixtral_all_params.json Move to moe-kernels package and switch to common MoE layer 2024-09-16 10:57:44 +00:00
test_flash_mixtral_load.json Add tests for Mixtral (#2520) 2024-09-16 12:39:18 +02:00
test_flash_mixtral.json Add tests for Mixtral (#2520) 2024-09-16 12:39:18 +02:00