hf_text-generation-inference/server/text_generation_server/layers/moe
Daniël de Kok 1c84a30fe6
MoE Marlin: support `desc_act` for `groupsize != -1` (#2590)
This change uses the updated Marlin MoE kernel from vLLM to support
MoE with activation sorting and groups.
2024-09-30 19:40:25 +02:00
..
__init__.py MoE Marlin: support `desc_act` for `groupsize != -1` (#2590) 2024-09-30 19:40:25 +02:00
fused_moe_rocm.py Update ROCM libs and improvements (#2579) 2024-09-30 10:54:32 +02:00
gptq_marlin.py MoE Marlin: support `desc_act` for `groupsize != -1` (#2590) 2024-09-30 19:40:25 +02:00
unquantized.py Update ROCM libs and improvements (#2579) 2024-09-30 10:54:32 +02:00