huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-24 00:12:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,101 Commits 229 Branches 59 Tags 23 MiB
git_v2.4.0
Commit Graph

2 Commits

Author SHA1 Message Date
Daniël de Kok
ce85efa968
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-17 18:08:58 +02:00
Daniël de Kok
7774655297
Add tests for Mixtral (#2520) 
Disable by default because CI runners do not have enough GPUs.
 2024-09-16 12:39:18 +02:00