Skip to content

Commit c36e21b

Browse files
mgoinAlvant
authored andcommitted
Fix the AMD weight loading tests (vllm-project#8390)
Signed-off-by: Alvant <[email protected]>
1 parent b3b2136 commit c36e21b

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)