Skip to content

Commit 2daba37

Browse files
Lijiachen1018lijiachen19
andauthored
Fix(patch): fix patch for vllm-ascend (#433)
Fix(patch): fix patch for vllm-ascend volcengine/verl#2564 Co-authored-by: lijiachen19 <[email protected]>
1 parent 8441e91 commit 2daba37

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ucm/integration/vllm/patch/patch_funcs/v092/vllm_ascend_patch.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,8 @@ def _enable_sparse() -> bool:
4040
def _apply_ascend_patch() -> None:
4141
"""Apply patch for vLLM-Ascend."""
4242
try:
43+
from vllm_ascend.patch import platform, worker
44+
4345
if _enable_sparse():
4446
_patch_attention_v1()
4547
_patch_mla_v1()

0 commit comments

Comments
 (0)