mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-09 23:45:54 +08:00
Fix triton install condition on CPU (#16600)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
parent
b3f2fddd17
commit
9883a18859
@ -17,4 +17,4 @@ torchvision==0.21.0; platform_machine == "ppc64le"
|
||||
datasets # for benchmark scripts
|
||||
|
||||
# cpu cannot use triton 3.3.0
|
||||
triton==3.2.0; platform_machine != "ppc64le"
|
||||
triton==3.2.0; platform_machine == "x86_64"
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user