2024-04-05 00:39:17 -07:00
|
|
|
# Common dependencies
|
2025-03-08 17:44:35 +01:00
|
|
|
-r common.txt
|
2024-04-05 00:39:17 -07:00
|
|
|
|
2025-12-18 14:36:49 +08:00
|
|
|
setuptools==77.0.3 # this version can reuse CMake build dir
|
|
|
|
|
|
2025-10-08 18:40:42 +01:00
|
|
|
numba == 0.61.2; platform_machine != "s390x" # Required for N-gram speculative decoding
|
2025-06-04 09:43:01 +08:00
|
|
|
|
2024-11-26 08:02:39 +05:30
|
|
|
# Dependencies for CPUs
|
2026-01-23 13:13:06 +00:00
|
|
|
torch==2.10.0+cpu; platform_machine == "x86_64" or platform_machine == "s390x"
|
2026-03-11 12:51:39 +08:00
|
|
|
torch==2.10.0; platform_machine == "aarch64" or platform_system == "Darwin" or platform_machine == "ppc64le" or platform_machine == "riscv64"
|
2025-01-31 02:59:39 +05:30
|
|
|
|
|
|
|
|
# required for the image processor of minicpm-o-2_6, this must be updated alongside torch
|
2026-03-11 12:51:39 +08:00
|
|
|
torchaudio; platform_machine != "s390x" and platform_machine != "riscv64"
|
2025-01-31 02:59:39 +05:30
|
|
|
|
|
|
|
|
# required for the image processor of phi3v, this must be updated alongside torch
|
2026-03-11 12:51:39 +08:00
|
|
|
torchvision; platform_machine != "s390x" and platform_machine != "riscv64"
|
2025-04-09 21:35:00 -07:00
|
|
|
|
2025-05-23 19:38:42 +08:00
|
|
|
# Intel Extension for PyTorch, only for x86_64 CPUs
|
2025-05-27 16:03:56 +08:00
|
|
|
intel-openmp==2024.2.1; platform_machine == "x86_64"
|
2025-07-25 22:33:56 +08:00
|
|
|
|
|
|
|
|
# Use this to gather CPU info and optimize based on ARM Neoverse cores
|
|
|
|
|
py-cpuinfo; platform_machine == "aarch64"
|
