From 1576347892b5c7eb3b55f22c17110efe769f1cea Mon Sep 17 00:00:00 2001 From: Shaojun Liu <61072813+liu-shaojun@users.noreply.github.com> Date: Tue, 20 May 2025 16:41:13 +0800 Subject: [PATCH] Update Dockerfile (#13168) --- docker/llm/serving/xpu/docker/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/llm/serving/xpu/docker/Dockerfile b/docker/llm/serving/xpu/docker/Dockerfile index abaa99e6..cc441f04 100644 --- a/docker/llm/serving/xpu/docker/Dockerfile +++ b/docker/llm/serving/xpu/docker/Dockerfile @@ -53,7 +53,7 @@ RUN set -eux && \ dpkg -i *.deb && \ # # Install Intel PyTorch extension for LLM inference - pip install --pre --upgrade ipex-llm[xpu_2.6] --extra-index-url https://download.pytorch.org/whl/xpu && \ + pip install --pre --upgrade 'ipex-llm[xpu_2.6]>=2.3.0b0,<2.3.0rc1' --extra-index-url https://download.pytorch.org/whl/xpu && \ pip install intel-extension-for-pytorch==2.6.10+xpu --extra-index-url=https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ && \ # # Build torch-ccl @@ -124,7 +124,7 @@ RUN set -eux && \ wget https://bootstrap.pypa.io/get-pip.py -O get-pip.py && \ python3 get-pip.py && rm get-pip.py && \ pip install --upgrade requests argparse urllib3 && \ - pip install --pre --upgrade ipex-llm[xpu_2.6] --extra-index-url https://download.pytorch.org/whl/xpu && \ + pip install --pre --upgrade 'ipex-llm[xpu_2.6]>=2.3.0b0,<2.3.0rc1' --extra-index-url https://download.pytorch.org/whl/xpu && \ pip install transformers_stream_generator einops tiktoken librosa && \ pip install --upgrade colorama && \ #