Skip to content

Commit a1a28cf

Browse files
hmellordbyoung18
authored andcommitted
Fix triton install condition on CPU (vllm-project#16600)
Signed-off-by: Harry Mellor <[email protected]>
1 parent 77b5dd1 commit a1a28cf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

requirements/cpu.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@ torchaudio; platform_machine != "ppc64le" and platform_machine != "s390x"
1212
torchaudio==2.6.0; platform_machine == "ppc64le"
1313

1414
# required for the image processor of phi3v, this must be updated alongside torch
15-
torchvision; platform_machine != "ppc64le" and platform_machine != "s390x"
15+
torchvision; platform_machine != "ppc64le" and platform_machine != "s390x"
1616
torchvision==0.21.0; platform_machine == "ppc64le"
1717
datasets # for benchmark scripts
1818

1919
# cpu cannot use triton 3.3.0
20-
triton==3.2.0; platform_machine != "ppc64le"
20+
triton==3.2.0; platform_machine == "x86_64"

0 commit comments

Comments
 (0)