diff --git a/python/sglang/srt/models/afmoe.py b/python/sglang/srt/models/afmoe.py index 543c8bb12..efdfaade7 100644 --- a/python/sglang/srt/models/afmoe.py +++ b/python/sglang/srt/models/afmoe.py @@ -62,9 +62,7 @@ from sglang.srt.utils import add_prefix, is_npu _is_npu = is_npu() -if not _is_npu: - from sglang.srt.layers.moe.fused_moe_triton import fused_moe -else: +if _is_npu: from sglang.srt.hardware_backend.npu.quantization.fused_moe_method_npu import ( fused_moe_npu as fused_moe, )