Skip to content

Commit 0fde7da

Browse files
committed
Dockerfile.rocm.ubi: also include audio,video,tensorizer extras for last stage
1 parent 25ee230 commit 0fde7da

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Dockerfile.ubi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ USER root
218218
RUN --mount=type=cache,target=/root/.cache/pip \
219219
--mount=type=cache,target=/root/.cache/uv \
220220
--mount=type=bind,from=build,src=/workspace/dist,target=/workspace/dist \
221-
HOME=/root uv pip install "$(echo /workspace/dist/*.whl)[tensorizer]" vllm-tgis-adapter==0.6.3
221+
HOME=/root uv pip install "$(echo /workspace/dist/*.whl)[audio,video,tensorizer]" vllm-tgis-adapter==0.6.3
222222

223223
ENV GRPC_PORT=8033 \
224224
PORT=8000 \

0 commit comments

Comments
 (0)