Skip to content

Commit 0f26017

Browse files
committed
lint fix
1 parent 942eb97 commit 0f26017

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/worker/model_runner_v1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1146,7 +1146,7 @@ def _calc_spec_decode_metadata(
11461146

11471147
# Compute the logits indices.
11481148
# [4, 1, 3, 1, 2]
1149-
num_sampled_tokens = num_draft_tokens + 1
1149+
num_sampled_tokens: np.ndarray = num_draft_tokens + 1
11501150
# Step 1. [4, 5, 8, 9, 11]
11511151
cu_num_sampled_tokens = np.cumsum(num_sampled_tokens, dtype=np.int32)
11521152
total_num_sampled_tokens = cu_num_sampled_tokens[-1]

0 commit comments

Comments
 (0)