at master 733 B view raw
1diff --git a/requirements/cpu.txt b/requirements/cpu.txt 2index d80354342..7434f32f0 100644 3--- a/requirements/cpu.txt 4+++ b/requirements/cpu.txt 5@@ -21,7 +21,4 @@ torchvision; platform_machine != "ppc64le" and platform_machine != "s390x" 6 torchvision==0.22.0; platform_machine == "ppc64le" 7 datasets # for benchmark scripts 8 9-# Intel Extension for PyTorch, only for x86_64 CPUs 10-intel-openmp==2024.2.1; platform_machine == "x86_64" 11-intel_extension_for_pytorch==2.6.0; platform_machine == "x86_64" # torch>2.6.0+cpu has performance regression on x86 platform, see https://github.com/pytorch/pytorch/pull/151218 12 triton==3.2.0; platform_machine == "x86_64" # Triton is required for torch 2.6+cpu, as it is imported in torch.compile.