Skip to content

Commit 79bda3a

Browse files
committed
fix mc2 issue
Signed-off-by: ganyi <pleaplusone.gy@gmail.com>
1 parent cb24491 commit 79bda3a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/models/deepseek_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ def forward(
314314
is_prefill = is_prefill or attn_metadata.with_prefill_across_dp
315315
# If this node is kv_consumer, we force the moe always runs in decode path to make sure
316316
# the behaviour aligned between dummy_run and normal model_execute.
317-
if self.kv_consumer is not None:
317+
if self.kv_consumer:
318318
is_prefill = False
319319
enable_force_load_balance = False
320320

0 commit comments

Comments
 (0)