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
|
|
|
|
2024-11-26 08:02:39 +05:30
|
|
|
# Dependencies for CPUs
|
2025-03-12 18:41:13 +08:00
|
|
|
torch==2.6.0+cpu; platform_machine == "x86_64"
|
2025-03-20 09:22:40 +01:00
|
|
|
torch==2.6.0; platform_system == "Darwin"
|
2025-03-25 15:45:14 +05:30
|
|
|
torch==2.6.0; platform_machine == "ppc64le" or platform_machine == "aarch64"
|
2025-03-06 22:10:53 +05:30
|
|
|
torch==2.7.0.dev20250304; platform_machine == "s390x"
|
2025-01-31 02:59:39 +05:30
|
|
|
|
|
|
|
|
# required for the image processor of minicpm-o-2_6, this must be updated alongside torch
|
2025-03-06 22:10:53 +05:30
|
|
|
torchaudio; platform_machine != "ppc64le" and platform_machine != "s390x"
|
2025-03-25 15:45:14 +05:30
|
|
|
torchaudio==2.6.0; platform_machine == "ppc64le"
|
2025-01-31 02:59:39 +05:30
|
|
|
|
|
|
|
|
# required for the image processor of phi3v, this must be updated alongside torch
|
2025-04-14 18:06:01 +01:00
|
|
|
torchvision; platform_machine != "ppc64le" and platform_machine != "s390x"
|
2025-03-25 15:45:14 +05:30
|
|
|
torchvision==0.21.0; platform_machine == "ppc64le"
|
2025-01-08 05:35:49 -03:00
|
|
|
datasets # for benchmark scripts
|
2025-04-09 21:35:00 -07:00
|
|
|
|
|
|
|
|
# cpu cannot use triton 3.3.0
|
2025-04-14 18:06:01 +01:00
|
|
|
triton==3.2.0; platform_machine == "x86_64"
|
