From c0d00f5be6d3ed390534dd909c82b639baf2d359 Mon Sep 17 00:00:00 2001 From: Xudong Zhang Date: Fri, 28 Jul 2023 14:37:40 +0800 Subject: [PATCH] [Fix] fix import error of RayWorker (#604) (#605) --- vllm/engine/ray_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vllm/engine/ray_utils.py b/vllm/engine/ray_utils.py index 06ebf207..f085f922 100644 --- a/vllm/engine/ray_utils.py +++ b/vllm/engine/ray_utils.py @@ -27,6 +27,7 @@ try: except ImportError: ray = None TorchDistributedWorker = None + RayWorker = None # pylint: disable=invalid-name if TYPE_CHECKING: from ray.util.placement_group import PlacementGroup