Skip to content

Commit eba18b2

Browse files
committed
make FusedMoEModularKernel a Leaf
Signed-off-by: Bill Nell <bnell@redhat.com>
1 parent fd999f2 commit eba18b2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/model_executor/layers/fused_moe/modular_kernel.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -348,8 +348,8 @@ def __init__(
348348
super().__init__()
349349
self.prepare_finalize = prepare_finalize
350350
self.fused_experts = fused_experts
351-
assert prepare_finalize.activation_format == fused_experts.activation_formats[ # noqa: E501
352-
0]
351+
assert prepare_finalize.activation_format == \
352+
fused_experts.activation_formats[0]
353353

354354
def forward(
355355
self,

0 commit comments

Comments
 (0)