diff --git a/requirements/cpu.txt b/requirements/cpu.txt index 1205c098e6f0..d845fb201cef 100644 --- a/requirements/cpu.txt +++ b/requirements/cpu.txt @@ -17,4 +17,4 @@ torchvision==0.21.0; platform_machine == "ppc64le" datasets # for benchmark scripts # cpu cannot use triton 3.3.0 -triton==3.2.0 +triton==3.2.0; platform_machine != "ppc64le"