Skip to content

Commit 11c3338

Browse files
committed
fix
Signed-off-by: xuyexiong <xuyexiong@huawei.com>
1 parent d015941 commit 11c3338

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm_ascend/spec_decode/mtp_proposer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -441,8 +441,8 @@ def _propose(
441441
aclgraph_runtime_mode, batch_descriptor = \
442442
self.runner.aclgraph_dispatcher.dispatch(batch_descriptor)
443443
if aclgraph_runtime_mode not in [
444-
CUDAGraphMode.PIECEWISE, CUDAGraphMode.NONE
445-
]:
444+
CUDAGraphMode.PIECEWISE, CUDAGraphMode.NONE
445+
]:
446446
# Fallback to piecewise graph, when acl full graph is enabled
447447
logger.debug(
448448
"Currently the eagle proposer only supports cudagraph_mode "

0 commit comments

Comments
 (0)