text-generation-inference
ce85efa9 - Move to moe-kernels package and switch to common MoE layer (#2511)

Commit
1 year ago
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
Author
Parents
Loading