huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-20 14:22:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
git_v2.4.0
text-generation-inference/server/text_generation_server/layers/moe
History
Daniël de Kok 64142489b6
Add support for fused MoE Marlin for AWQ (#2616) 
* Add support for fused MoE Marlin for AWQ

This uses the updated MoE Marlin kernels from vLLM.

* Add integration test for AWQ MoE
2024-10-08 11:56:41 +02:00
..
__init__.py Add support for fused MoE Marlin for AWQ (#2616) 2024-10-08 11:56:41 +02:00
fused_moe_rocm.py Update ROCM libs and improvements (#2579) 2024-09-30 10:54:32 +02:00
gptq_marlin.py Add support for fused MoE Marlin for AWQ (#2616) 2024-10-08 11:56:41 +02:00
unquantized.py Update ROCM libs and improvements (#2579) 2024-09-30 10:54:32 +02:00