From 736be0711cbd076a53aa3e049fe0f397233bb0b6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Daniele=20Trifir=C3=B2?= Date: Thu, 27 Mar 2025 17:03:42 +0100 Subject: [PATCH] Dockerfile.rocm.ubi: also include audio,video,tensorizer extras for last stage --- Dockerfile.ubi | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.ubi b/Dockerfile.ubi index cd17fcb9e04d..70a8dc8935ec 100644 --- a/Dockerfile.ubi +++ b/Dockerfile.ubi @@ -218,7 +218,7 @@ USER root RUN --mount=type=cache,target=/root/.cache/pip \ --mount=type=cache,target=/root/.cache/uv \ --mount=type=bind,from=build,src=/workspace/dist,target=/workspace/dist \ - HOME=/root uv pip install "$(echo /workspace/dist/*.whl)[tensorizer]" vllm-tgis-adapter==0.6.3 + HOME=/root uv pip install "$(echo /workspace/dist/*.whl)[audio,video,tensorizer]" vllm-tgis-adapter==0.6.3 ENV GRPC_PORT=8033 \ PORT=8000 \