fix optimum-intel

Signed-off-by: changwangss <[email protected]>
intel · Jul 5, 2024 · 8006e59 · 8006e59
1 parent 590024d
commit 8006e59
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/examples/huggingface/pytorch/code-generation/quantization/Dockerfile-multiple b/examples/huggingface/pytorch/code-generation/quantization/Dockerfile-multiple
@@ -61,7 +61,7 @@ ENV COMPOSE_DOCKER_CLI_BUILD=0
 # Install torch and intel-extension-for-pytorch 2.1
 RUN python3 -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
 RUN python3 -m pip install intel-extension-for-pytorch intel-extension-for-transformers optimum
-RUN python3 -m pip install git+https://github.com/huggingface/optimum-intel.git@f95dea1ae8966dee4d75d622e7b2468c514ba02d
+RUN python3 -m pip install git+https://github.com/huggingface/optimum-intel.git@50d867c13b22c22eda451ddb67bddb8159670f85
 RUN python3 -m pip install git+https://github.com/bigcode-project/bigcode-evaluation-harness@0d84db85f9ff971fa23a187a3347b7f59af288dc
 
 # Standard requirements

diff --git a/examples/huggingface/pytorch/code-generation/quantization/requirements.txt b/examples/huggingface/pytorch/code-generation/quantization/requirements.txt
@@ -10,6 +10,6 @@ transformers >= 4.35.0
 tiktoken #code_gen
 neural-compressor
 intel_extension_for_pytorch==2.3.0
-optimum-intel
+git+https://github.com/huggingface/optimum-intel.git@50d867c13b22c22eda451ddb67bddb8159670f85
 auto-round==0.2
 git+https://github.com/bigcode-project/bigcode-evaluation-harness@094c7cc197d13a53c19303865e2056f1c7488ac1
diff --git a/examples/huggingface/pytorch/text-generation/quantization/requirements_GPU.txt b/examples/huggingface/pytorch/text-generation/quantization/requirements_GPU.txt
@@ -7,7 +7,7 @@ sentencepiece != 0.1.92
 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
 torch==2.1.0a0
 transformers
-optimum-intel
+git+https://github.com/huggingface/optimum-intel.git@50d867c13b22c22eda451ddb67bddb8159670f85
 bitsandbytes  #baichuan
 transformers_stream_generator
 tiktoken  #qwen

diff --git a/examples/huggingface/pytorch/text-generation/quantization/requirements_sq.txt b/examples/huggingface/pytorch/text-generation/quantization/requirements_sq.txt
@@ -7,7 +7,7 @@ sentencepiece != 0.1.92
 torch==2.3.0+cpu
 transformers==4.38.1
 intel_extension_for_pytorch==2.3.0
-optimum-intel
+git+https://github.com/huggingface/optimum-intel.git@50d867c13b22c22eda451ddb67bddb8159670f85
 bitsandbytes  #baichuan
 transformers_stream_generator
 tiktoken  #qwen

diff --git a/tests/requirements.txt b/tests/requirements.txt
@@ -7,6 +7,7 @@ datasets==2.16.1
 einops
 evaluate
 gguf
+git+https://github.com/huggingface/optimum-intel.git@50d867c13b22c22eda451ddb67bddb8159670f85
 git+https://github.com/intel/neural-compressor.git
 git+https://github.com/intel/neural-speed.git
 intel-extension-for-pytorch==2.3.0
@@ -16,7 +17,6 @@ mlflow
 nlpaug==1.1.9
 onnx
 onnxruntime
-optimum-intel
 peft==0.6.2
 py-cpuinfo
 sacremoses