File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 33
44# Parameterize base image components for RHEL 9.4/9.6 and TencentOS 3.1
55ARG DOCKER_URL=vault.habana.ai/gaudi-docker
6- ARG VERSION=1.21.2
7- ARG BASE_NAME=rhel9.2
6+ ARG VERSION=1.21.5
7+ ARG BASE_NAME=rhel9.4
88ARG PT_VERSION=2.6.0
99ARG REVISION=latest
1010ARG REPO_TYPE=habanalabs
@@ -45,7 +45,7 @@ RUN mkdir -p $VLLM_PATH && \
4545 git remote add upstream https://github.com/vllm-project/vllm.git && \
4646 git fetch upstream --tags || true && \
4747 git checkout ${VLLM_PROJECT_COMMIT} && \
48- bash -c " pip install -r <(sed '/^[ torch] /d' requirements/build.txt)" && \
48+ pip install -r <(sed '/^torch/d' requirements/build.txt) && \
4949 VLLM_TARGET_DEVICE=empty pip install --no-build-isolation -e .
5050
5151# Clone the vllm-gaudi repository and install inside the container
@@ -55,7 +55,7 @@ RUN mkdir -p $VLLM_PATH2 && \
5555 git checkout ${VLLM_GAUDI_COMMIT} && \
5656 VLLM_TARGET_DEVICE=hpu && pip install -v -e $VLLM_PATH2
5757
58- # to be enabled later PWolsza
58+ # to be verified later PWolsza
5959# RUN pip3 install -v -e $VLLM_PATH/tests/vllm_test_utils
6060
6161# Install additional Python packages
You can’t perform that action at this time.
0 commit comments