Add intel-extension-for-pytorch to Transformers-like API requirements (…

…#2113) Signed-off-by: Sun, Xuehao <[email protected]>
intel · Feb 8, 2025 · 9c3d4a1 · 9c3d4a1
1 parent a617115
commit 9c3d4a1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/...e-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt b/...e-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt
@@ -15,3 +15,6 @@ lm-eval==0.4.3
 huggingface_hub
 numba
 tbb
+intel-extension-for-pytorch==2.4.0
+--extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/
+oneccl_bind_pt==2.4.0