"vllm/model_executor/layers/attention/attention.py" did not exist on "43721bc67f9f30dcd1c893af370727b835d3d6dd"
cpu.txt 856 Bytes
Newer Older
1
# Common dependencies
2
-r common.txt
3

4
# Dependencies for CPUs
汪志鹏's avatar
汪志鹏 committed
5
6
packaging>=24.2
setuptools>=77.0.3,<80.0.0
Huy Do's avatar
Huy Do committed
7
8
9
10
--extra-index-url https://download.pytorch.org/whl/cpu
torch==2.7.0+cpu; platform_machine == "x86_64"
torch==2.7.0; platform_system == "Darwin"
torch==2.7.0; platform_machine == "ppc64le" or platform_machine == "aarch64"
11
12

# required for the image processor of minicpm-o-2_6, this must be updated alongside torch
13
torchaudio; platform_machine != "ppc64le" and platform_machine != "s390x"
Huy Do's avatar
Huy Do committed
14
torchaudio==2.7.0; platform_machine == "ppc64le"
15
16

# required for the image processor of phi3v, this must be updated alongside torch
17
torchvision; platform_machine != "ppc64le" and platform_machine != "s390x"
Huy Do's avatar
Huy Do committed
18
torchvision==0.22.0; platform_machine == "ppc64le"
19
datasets # for benchmark scripts
20
21

# cpu cannot use triton 3.3.0
22
triton==3.2.0; platform_machine == "x86_64"