text-generation-inference

huggingface/text-generation-inference

Fork 0

mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-21 06:42:10 +00:00

Commit Graph

Author SHA1 Message Date

Author	SHA1	Message	Date
Daniël de Kok	3f14cd1420	Add `DenseMoELayer` and wire it up in Mixtral/Deepseek V2 (#2537 ) This replaces the custom layers in both models.	2024-09-24 14:27:06 +02:00
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

3f14cd1420

Add DenseMoELayer and wire it up in Mixtral/Deepseek V2 (#2537 )

This replaces the custom layers in both models.

2024-09-24 14:27:06 +02:00

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

2 Commits