chore(vllm-tensorizer): Update naming for base images

Author: JustinPerlman

.github/configurations/vllm-tensorizer.yml

@@ -2,7 +2,7 @@ vllm-commit:
   - 'b6553be1bc75f046b00046a4ad7576364d03c835'
 flashinfer-commit:
   - 'v0.2.6.post1'
-base-image:
+builder-base-image:
   - 'ghcr.io/coreweave/ml-containers/torch-extras:es-cuda-12.9.1-74755e9-nccl-cuda12.9.1-ubuntu22.04-nccl2.27.5-1-torch2.7.1-vision0.22.1-audio2.7.1-abi1'
-lean-base-image:
+final-base-image:
   - 'ghcr.io/coreweave/ml-containers/torch-extras:es-cuda-12.9.1-74755e9-base-cuda12.9.1-ubuntu22.04-torch2.7.1-vision0.22.1-audio2.7.1-abi1'

.github/workflows/vllm-tensorizer.yml

@@ -25,5 +25,5 @@ jobs:
       build-args: |
         VLLM_COMMIT=${{ matrix.vllm-commit }}
         FLASHINFER_COMMIT=${{ matrix.flashinfer-commit }}
-        BASE_IMAGE=${{ matrix.base-image }}
-        LEAN_BASE_IMAGE=${{ matrix.lean-base-image }}
+        BASE_IMAGE=${{ matrix.builder-base-image }}
+        LEAN_BASE_IMAGE=${{ matrix.final-base-image }}

vllm-tensorizer/Dockerfile

@@ -1,11 +1,11 @@
-ARG BASE_IMAGE="ghcr.io/coreweave/ml-containers/torch-extras:es-cuda-12.9.1-74755e9-nccl-cuda12.9.1-ubuntu22.04-nccl2.27.5-1-torch2.7.1-vision0.22.1-audio2.7.1-abi1"
-ARG LEAN_BASE_IMAGE="ghcr.io/coreweave/ml-containers/torch-extras:es-cuda-12.9.1-74755e9-base-cuda12.9.1-ubuntu22.04-torch2.7.1-vision0.22.1-audio2.7.1-abi1"
+ARG BUILDER_BASE_IMAGE="ghcr.io/coreweave/ml-containers/torch-extras:es-cuda-12.9.1-74755e9-nccl-cuda12.9.1-ubuntu22.04-nccl2.27.5-1-torch2.7.1-vision0.22.1-audio2.7.1-abi1"
+ARG FINAL_BASE_IMAGE="ghcr.io/coreweave/ml-containers/torch-extras:es-cuda-12.9.1-74755e9-base-cuda12.9.1-ubuntu22.04-torch2.7.1-vision0.22.1-audio2.7.1-abi1"
 
 FROM scratch AS freezer
 WORKDIR /
 COPY --chmod=755 freeze.sh /
 
-FROM ${BASE_IMAGE} AS builder-base
+FROM ${BUILDER_BASE_IMAGE} AS builder-base
 
 ARG MAX_JOBS="32"
 
@@ -83,7 +83,7 @@ RUN --mount=type=bind,from=flashinfer-downloader,source=/git/flashinfer,target=/
 WORKDIR /wheels
 
 
-FROM ${LEAN_BASE_IMAGE} AS base
+FROM ${FINAL_BASE_IMAGE} AS base
 
 WORKDIR /workspace