mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-09 17:55:01 +08:00
9368cc90b2
Signed-off-by: Tsai, Louie <louie.tsai@intel.com> Co-authored-by: Li, Jiang <bigpyj64@gmail.com>
32 lines
1.2 KiB
Plaintext
32 lines
1.2 KiB
Plaintext
# Common dependencies
|
|
-r common.txt
|
|
|
|
numba == 0.60.0; python_version == '3.9' # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding
|
|
numba == 0.61.2; python_version > '3.9'
|
|
|
|
# Dependencies for CPUs
|
|
packaging>=24.2
|
|
setuptools>=77.0.3,<80.0.0
|
|
--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"
|
|
|
|
# required for the image processor of minicpm-o-2_6, this must be updated alongside torch
|
|
torchaudio; platform_machine != "ppc64le" and platform_machine != "s390x"
|
|
torchaudio==2.7.0; platform_machine == "ppc64le"
|
|
|
|
# required for the image processor of phi3v, this must be updated alongside torch
|
|
torchvision; platform_machine != "ppc64le" and platform_machine != "s390x"
|
|
torchvision==0.22.0; platform_machine == "ppc64le"
|
|
datasets # for benchmark scripts
|
|
|
|
# cpu cannot use triton 3.3.0
|
|
triton==3.2.0; platform_machine == "x86_64"
|
|
|
|
# Intel Extension for PyTorch, only for x86_64 CPUs
|
|
intel-openmp==2024.2.1; platform_machine == "x86_64"
|
|
intel_extension_for_pytorch==2.7.0; platform_machine == "x86_64"
|
|
py-libnuma; platform_system != "Darwin"
|
|
psutil; platform_system != "Darwin"
|