Skip to content

Commit 8ac3f51

Browse files
committed
fix mypy
Signed-off-by: Angazenn <supperccell@163.com>
1 parent 5565a33 commit 8ac3f51

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/attention/attention_v1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -459,7 +459,7 @@ def _forward_v1_style(
459459
if torch.version.cann.startswith("8.3"):
460460
# TODO:The npu_fused_infer_attention_score op is planned to
461461
# be utilized in a wider range in upcoming versions.
462-
num_block, block_size, head_num, head_dim = self.key_cache.shape
462+
num_block, block_size, _, _ = self.key_cache.shape # type: ignore
463463
key = self.key_cache.view( # type: ignore
464464
num_block, block_size, -1)
465465
value = self.value_cache.view( # type: ignore

0 commit comments

Comments
 (0)