Skip to content

Commit b979ee3

Browse files
authored
[Misc] Code clean up (#1679)
Make model_runner_v1 more readable - vLLM version: v0.9.2 - vLLM main: vllm-project/vllm@baed180 Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
1 parent 392fd72 commit b979ee3

File tree

2 files changed

+138
-254
lines changed

2 files changed

+138
-254
lines changed

vllm_ascend/worker/eagle_proposer_v1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from vllm.model_executor.models.llama_eagle3 import Eagle3LlamaForCausalLM
1515
from vllm.v1.sample.metadata import SamplingMetadata
1616

17-
from vllm_ascend.attention.attention import AttentionMaskBuilder
17+
from vllm_ascend.attention.attention_mask import AttentionMaskBuilder
1818
from vllm_ascend.attention.attention_v1 import AscendAttentionState
1919

2020
logger = init_logger(__name__)

0 commit comments

Comments
 (0)