xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-14 00:05:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] Upgrade triton to 2.2.0 (#4061)

Browse Source
This commit is contained in:
Roy 2024-04-15 08:43:54 +08:00 committed by GitHub
parent aceb17cf2d
commit 8db1bf32f8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements-cpu.txt
View File

@ -3,4 +3,4 @@
# Dependencies for x86_64 CPUs # Dependencies for x86_64 CPUs
torch == 2.2.1+cpu torch == 2.2.1+cpu
triton >= 2.1.0 # FIXME(woosuk): This is a hack to avoid import error. triton >= 2.2.0 # FIXME(woosuk): This is a hack to avoid import error.

1
requirements-cuda.txt
View File

@ -7,4 +7,3 @@ pynvml == 11.5.0
vllm-nccl-cu12>=2.18,<2.19 # for downloading nccl library vllm-nccl-cu12>=2.18,<2.19 # for downloading nccl library
torch == 2.2.1 torch == 2.2.1
xformers == 0.0.25 # Requires PyTorch 2.2.1 xformers == 0.0.25 # Requires PyTorch 2.2.1
triton >= 2.1.0