From 9d504109c85192e9ae4c450a26ab3a09c60467fe Mon Sep 17 00:00:00 2001 From: "Sun, Xuehao" Date: Fri, 24 Jan 2025 15:46:03 +0800 Subject: [PATCH 1/2] Add intel-extension-for-pytorch Signed-off-by: Sun, Xuehao --- .../weight_only/text-generation/requirements_cpu_woq.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt b/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt index 06b8ce32dda..0a5c0f83d51 100644 --- a/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt +++ b/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt @@ -15,3 +15,4 @@ lm-eval==0.4.3 huggingface_hub numba tbb +intel-extension-for-pytorch From d7d6a8848359d925fc4d9cf1f1c9e8cdbf4ec396 Mon Sep 17 00:00:00 2001 From: "Sun, Xuehao" Date: Fri, 24 Jan 2025 16:38:34 +0800 Subject: [PATCH 2/2] Update requirements with specific versions for intel-extension-for-pytorch and oneccl_bind_pt Signed-off-by: Sun, Xuehao --- .../weight_only/text-generation/requirements_cpu_woq.txt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt b/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt index 0a5c0f83d51..a3e0a5933e7 100644 --- a/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt +++ b/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt @@ -15,4 +15,6 @@ lm-eval==0.4.3 huggingface_hub numba tbb -intel-extension-for-pytorch +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