Skip to content

Commit 72a2bf3

Browse files
remove reserved_mc2_mask
Signed-off-by: SunnyLee219 <3294305115@qq.com>
1 parent 2cf8e28 commit 72a2bf3

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

vllm_ascend/worker/mtp_proposer_v1.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,9 +48,6 @@ def __init__(
4848
device=self.runner.device)
4949
self.torchair_compiled_model = None # type: ignore
5050
self.torchair_compiled_models = {} # type: ignore
51-
self.reserved_mc2_mask = torch.zeros(512,
52-
dtype=torch.bool,
53-
device=self.runner.device)
5451
self.torchair_graph_enabled = get_ascend_config(
5552
).torchair_graph_config.enabled
5653

0 commit comments

Comments
 (0)