huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-05-20 09:12:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,190 Commits 232 Branches 60 Tags 28 MiB
51506aa57a
Commit Graph

2 Commits

Author SHA1 Message Date
Daniël de Kok
29a93b78ba Move to moe-kernels package and switch to common MoE layer (#2511) 
* 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.

* Make `cargo check` pass

* Update runner
 2024-09-25 06:18:05 +00:00
Daniël de Kok
e8c329372b Add tests for Mixtral (#2520) 
Disable by default because CI runners do not have enough GPUs.
 2024-09-25 06:16:08 +00:00