Fix vLLM ROCm build

Signed-off-by: Vaibhav Jain <[email protected]>

Author: vaibhavjainwiz

Dockerfile.rocm.ubi

@@ -1,7 +1,7 @@
 ## Global Args ##################################################################
 ARG BASE_UBI_IMAGE_TAG=9.5-1742914212
 ARG PYTHON_VERSION=3.12
-ARG VLLM_VERSION
+ARG VLLM_VERSION="v0.8.4"
 # Default ROCm ARCHes to build vLLM for.
 ARG PYTORCH_ROCM_ARCH="gfx908;gfx90a;gfx942;gfx1100"
 ARG MAX_JOBS=12
@@ -47,7 +47,7 @@ gpgkey=https://repo.radeon.com/rocm/rocm.gpg.key" > /etc/yum.repos.d/amdgpu.repo
 
 
 RUN --mount=type=cache,target=/root/.cache/uv \
-    export version="$(awk -F. '{print $1"."$2}' <<< $ROCM_VERSION)" && \
+    export version="$(awk -F. '{print $1"."$2}' <<< ${ROCM_VERSION})" && \
     uv pip install --pre \
         --index-url "https://download.pytorch.org/whl/rocm${version}" \
         torch==2.7.0+rocm${version}\