vllm/requirements-cpu.txt
Sanket Kale a6760f6456
[Feature] vLLM ARM Enablement for AARCH64 CPUs (#9228)
Signed-off-by: Sanket Kale <sanketk.kale@fujitsu.com>
Co-authored-by: Sanket Kale <sanketk.kale@fujitsu.com>
Co-authored-by: mgoin <michael@neuralmagic.com>
2024-11-25 18:32:39 -08:00

7 lines
326 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