Skip to content

Commit e49759d

Browse files
committed
Fix vllm build
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
1 parent d879360 commit e49759d

2 files changed

Lines changed: 3 additions & 4 deletions

File tree

.github/workflows/build-trtllm-vllm-image.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ on:
2525
description: Version of vLLM to use for the build
2626
required: true
2727
type: string
28-
default: 0.11.2
28+
default: v0.11.2
2929
trtllm_commit:
3030
description: Commit of TRT-LLM to use for the build
3131
required: true
@@ -66,7 +66,7 @@ env:
6666
container-registry: ${{ vars.container-registry || 'nemoci.azurecr.io' }}
6767
base-image-tag: ${{ inputs.base_image_tag || '25.11-py3' }}
6868
trtllm-commit: ${{ inputs.trtllm_commit || '48b7b5d8b7ed1e6bff57aecc6ff7d1533288bed8' }}
69-
vllm-version: ${{ inputs.vllm_version || '0.11.2' }}
69+
vllm-version: ${{ inputs.vllm_version || 'v0.11.2' }}
7070
trt-version: ${{ inputs.trt_version || '10.14.1.48' }}
7171
cuda-version: ${{ inputs.cuda_version || '13.0' }}
7272
cudnn-version: ${{ inputs.cudnn_version || '9.17.0.29-1' }}

docker/Dockerfile.trtllm_vllm

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,8 @@ WORKDIR /opt
2929

3030
FROM base AS vllm_wheel
3131

32-
ARG VLLM_VERSION=0.11.2
32+
ARG VLLM_VERSION
3333
ARG MAX_JOBS=4
34-
ENV VLLM_VERSION=$VLLM_VERSION
3534

3635
WORKDIR /src/vllm
3736

0 commit comments

Comments
 (0)