Skip to content

Commit fc94685

Browse files
committed
just 'if'
Signed-off-by: Ziliang Peng <[email protected]>
1 parent 73b5f43 commit fc94685

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/platforms/cuda.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def supported_dtypes(self) -> list[torch.dtype]:
6464
if self.has_device_capability(80):
6565
# Ampere and Hopper or later NVIDIA GPUs.
6666
return [torch.bfloat16, torch.float16, torch.float32]
67-
elif self.has_device_capability(60):
67+
if self.has_device_capability(60):
6868
# Pascal, Volta and Turing NVIDIA GPUs, BF16 is not supported
6969
return [torch.float16, torch.float32]
7070
# Kepler and Maxwell NVIDIA GPUs, only FP32 is supported,

0 commit comments

Comments
 (0)