Skip to content

Commit c9791f1

Browse files
authored
[BugFix] Fix broken import in initialize_ray_cluster() (#27838)
Signed-off-by: Nick Hill <nhill@redhat.com>
1 parent e7acb20 commit c9791f1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/v1/executor/ray_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -322,7 +322,7 @@ def initialize_ray_cluster(
322322

323323
# Prevalidate GPU requirements before Ray processing
324324
if current_platform.is_cuda() and parallel_config.world_size > 1:
325-
from vllm.utils import cuda_device_count_stateless
325+
from vllm.utils.torch_utils import cuda_device_count_stateless
326326

327327
available_gpus = cuda_device_count_stateless()
328328
if parallel_config.world_size > available_gpus:

0 commit comments

Comments
 (0)