Skip to content

Commit d232d49

Browse files
author
weijinqian
committed
handle clean code
Signed-off-by: weijinqian_v1 <weijinqian@huawei.com>
1 parent b97baf4 commit d232d49

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/multistream/ms_split.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ def model_input_split_v1_attn(
313313
# chunked prefill
314314
assert attn_metadata.attn_mask is not None
315315
if has_prefill_pre:
316-
attn_state_pre = attn_state_post = AscendAttentionState.ChunkedPrefill # noqa
316+
attn_state_pre = attn_state_post = AscendAttentionState.ChunkedPrefill # noqa
317317
attn_mask_pre = attn_metadata.attn_mask[:token_index, :max(
318318
seq_lens_pre)].contiguous()
319319
attn_state_post = AscendAttentionState.ChunkedPrefill # noqa

0 commit comments

Comments
 (0)