diff --git a/vllm/lora/layers/fused_moe.py b/vllm/lora/layers/fused_moe.py index 8f5cb14c3..be1fd7cdb 100644 --- a/vllm/lora/layers/fused_moe.py +++ b/vllm/lora/layers/fused_moe.py @@ -143,9 +143,7 @@ class FusedMoEWithLoRA(BaseLayerWithLoRA): m_fused_moe_fn.fused_experts, (MarlinExperts, UnfusedOAITritonExperts) ) else: - assert isinstance( - m_fused_moe_fn.fused_experts, (MarlinExperts, TritonExperts) - ) + assert isinstance(m_fused_moe_fn.fused_experts, TritonExperts) def fwd_decorator(layer, func): def wrapper(*args, **kwargs):