Skip to content

Commit 60c880b

Browse files
committed
fix the missing args
Signed-off-by: Lu Fang <[email protected]>
1 parent 0dcdd48 commit 60c880b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

vllm/model_executor/layers/fused_moe/fused_marlin_moe.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -731,6 +731,7 @@ def apply(
731731
if self.quant_config.use_int4_w4a16
732732
else scalar_types.float4_e2m1f.id
733733
), # works only for w4a16
734+
apply_router_weight_on_input=apply_router_weight_on_input,
734735
global_num_experts=global_num_experts,
735736
activation=activation,
736737
expert_map=expert_map,

0 commit comments

Comments
 (0)