Skip to content
This repository was archived by the owner on Sep 4, 2025. It is now read-only.

Commit b6c75e1

Browse files
authored
Fix the AMD weight loading tests (vllm-project#8390)
1 parent b71c956 commit b6c75e1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -883,7 +883,7 @@ def __init__(
883883
from vllm.executor import ray_utils
884884
backend = "mp"
885885
ray_found = ray_utils.ray_is_available()
886-
if (torch.cuda.is_available()
886+
if (current_platform.is_cuda()
887887
and cuda_device_count_stateless() < self.world_size):
888888
if not ray_found:
889889
raise ValueError("Unable to load Ray which is "

0 commit comments

Comments
 (0)