diff --git a/requirements/cpu.txt b/requirements/cpu.txt index d845fb201cef..69f732c2417a 100644 --- a/requirements/cpu.txt +++ b/requirements/cpu.txt @@ -12,9 +12,9 @@ torchaudio; platform_machine != "ppc64le" and platform_machine != "s390x" torchaudio==2.6.0; 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; platform_machine != "ppc64le" and platform_machine != "s390x" torchvision==0.21.0; platform_machine == "ppc64le" datasets # for benchmark scripts # cpu cannot use triton 3.3.0 -triton==3.2.0; platform_machine != "ppc64le" +triton==3.2.0; platform_machine == "x86_64"