Dockerfile.rocm.ubi: add [audio,video,tensorizer] extras to vllm install (#368)

https://issues.redhat.com/browse/INFERENG-375

Author: dtrifiro

Dockerfile.rocm.ubi

@@ -211,7 +211,7 @@ RUN --mount=type=bind,from=build_amdsmi,src=/install,target=/install/amdsmi/ \
         --extra-index-url "https://download.pytorch.org/whl/nightly/rocm${version}" \
         /install/amdsmi/*.whl\
         /install/flashattention/*.whl\
-        /install/vllm/*.whl
+        "$(echo /install/vllm/*.whl)[audio,video,tensorizer]"
 
 ENV HF_HUB_OFFLINE=1 \
     HOME=/home/vllm \
@@ -253,7 +253,10 @@ USER root
 RUN --mount=type=cache,target=/root/.cache/pip \
     --mount=type=cache,target=/root/.cache/uv \
     --mount=type=bind,from=build_vllm,src=/workspace/dist,target=/install/vllm/ \
-    HOME=/root uv pip install /install/vllm/*.whl vllm-tgis-adapter==0.6.3
+    HOME=/root uv pip install \
+        "$(echo /install/vllm/*.whl)[audio,video,tensorizer]" \
+        vllm-tgis-adapter==0.6.3
+
 
 ENV GRPC_PORT=8033 \
     PORT=8000 \