text-generation-inference/server/text_generation_server/layers/moe
Daniël de Kok e618ce3ada Fix: make moe_kernels imports conditional
`moe-kernels` is an optional dependency, so make sure that we run
without installing this package.

Fixes #2621.
2024-10-08 11:05:28 +00: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 Fix: make moe_kernels imports conditional 2024-10-08 11:05:28 +00:00
unquantized.py Fix: make moe_kernels imports conditional 2024-10-08 11:05:28 +00:00