ada19210a3
Signed-off-by: Dilip Gowda Bhagavan <dilip.bhagavan@ibm.com> Signed-off-by: Rishika Kedia <rishika.kedia@in.ibm.com> Co-authored-by: Rishika Kedia <rishika.kedia@in.ibm.com>
17 lines
839 B
Plaintext
17 lines
839 B
Plaintext
# Common dependencies
|
|
-r requirements-common.txt
|
|
|
|
# Dependencies for CPUs
|
|
torch==2.5.1+cpu; platform_machine != "ppc64le" and platform_machine != "aarch64" and platform_system != "Darwin" and platform_machine != "s390x"
|
|
torch==2.5.1; platform_machine == "ppc64le" or platform_machine == "aarch64" or platform_system == "Darwin"
|
|
torch==2.7.0.dev20250304; platform_machine == "s390x"
|
|
|
|
# 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.5.1; 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.20.1; platform_machine == "ppc64le"
|
|
datasets # for benchmark scripts
|