vllm/requirements-cpu.txt
Li, Jiang f93bf2b189
[Bugfix][CI][CPU] add missing datasets package to requirements-cpu.txt (#11159)
Signed-off-by: jiang1.li <jiang1.li@intel.com>
2024-12-13 08:50:35 +00:00

8 lines
359 B
Plaintext

# Common dependencies
-r requirements-common.txt
# Dependencies for CPUs
torch==2.5.1+cpu; platform_machine != "ppc64le" and platform_machine != "aarch64"
torch==2.5.1; platform_machine == "aarch64"
torchvision; platform_machine != "ppc64le" # required for the image processor of phi3v, this must be updated alongside torch
datasets # for benchmark scripts