diff --git a/docker/llm/inference/cpu/docker/Dockerfile b/docker/llm/inference/cpu/docker/Dockerfile index c8cb047f..9a3ba621 100644 --- a/docker/llm/inference/cpu/docker/Dockerfile +++ b/docker/llm/inference/cpu/docker/Dockerfile @@ -65,11 +65,8 @@ RUN cp /root/miniconda3/condabin/conda /usr/bin && \ pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cpu && \ pip install intel-extension-for-pytorch==2.2.0 && \ pip install oneccl_bind_pt==2.2.0 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/ && \ - cd ./intel-extension-for-pytorch && \ - pip install -r requirements.txt && \ pip install transformers==4.36.2 && \ pip install transformers_stream_generator && \ - cd /llm && \ echo "conda deactivate" >> /root/.bashrc ENTRYPOINT ["/bin/bash"]