Skip to content

Commit

Permalink
RayWorkerWrapper is not None
Browse files Browse the repository at this point in the history
Signed-off-by: Chenguang Li <757486878@qq.com>
  • Loading branch information
noemotiovon committed Feb 26, 2025
1 parent 839df78 commit 02b6c59
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion vllm_ascend/patch/ray_patch.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
from vllm.executor.ray_utils import RayWorkerWrapper

if RayWorkerWrapper is not None:

class NPURayWorkerWrapper(RayWorkerWrapper):
"""Importing torch_npu in other Ray processes through an empty class and
a monkey patch.
Expand All @@ -15,7 +16,7 @@ class NPURayWorkerWrapper(RayWorkerWrapper):
Worker process.
See https://github.com/vllm-project/vllm-ascend/pull/92.
"""
pass

pass

vllm.executor.ray_utils.RayWorkerWrapper = NPURayWorkerWrapper

0 comments on commit 02b6c59

Please sign in to comment.