Update vllm/model_executor/layers/fused_moe/aiter_experts.py

accepted suggestion

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
Signed-off-by: HakJu Kim <[email protected]>
Signed-off-by: HakJu Kim <[email protected]>

Author: whitememory

vllm/model_executor/layers/fused_moe/aiter_experts.py

@@ -111,7 +111,7 @@ def apply(
             activation=activation,
             apply_router_weight_on_input=apply_router_weight_on_input,
             expert_map=expert_map,
-            expert_num_tokens=expert_tokens_meta.expert_num_tokens,
+            expert_num_tokens=expert_tokens_meta.expert_num_tokens if expert_tokens_meta is not None else None,
             output_dtype=output.dtype,
             quant_config=self.quant_config,
             a1q_scale=a1q_scale,