File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed
Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change 2222from vllm .v1 .engine .detokenizer import Detokenizer
2323from vllm .v1 .engine .processor import Processor
2424from vllm .v1 .executor .abstract import Executor
25+ from vllm .v1 .executor .ray_utils import initialize_ray_cluster
2526
2627logger = init_logger (__name__ )
2728
@@ -131,7 +132,11 @@ def _get_executor_cls(cls, vllm_config: VllmConfig) -> Type[Executor]:
131132 executor_class : Type [Executor ]
132133 distributed_executor_backend = (
133134 vllm_config .parallel_config .distributed_executor_backend )
134- if distributed_executor_backend == "mp" :
135+ if distributed_executor_backend == "ray" :
136+ initialize_ray_cluster (vllm_config .parallel_config )
137+ from vllm .v1 .executor .ray_executor import RayExecutor
138+ executor_class = RayExecutor
139+ elif distributed_executor_backend == "mp" :
135140 from vllm .v1 .executor .multiproc_executor import MultiprocExecutor
136141 executor_class = MultiprocExecutor
137142 else :
You can’t perform that action at this time.
0 commit comments