xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-15 02:05:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Fix] fix import error of RayWorker (#604) (#605)

Browse Source
This commit is contained in:
Xudong Zhang 2023-07-28 14:37:40 +08:00 committed by GitHub
parent 58a072be15
commit c0d00f5be6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
vllm/engine/ray_utils.py
View File

@ -27,6 +27,7 @@ try:
except ImportError: except ImportError:
ray = None ray = None
TorchDistributedWorker = None TorchDistributedWorker = None
RayWorker = None # pylint: disable=invalid-name
if TYPE_CHECKING: if TYPE_CHECKING:
from ray.util.placement_group import PlacementGroup from ray.util.placement_group import PlacementGroup