misc(gha): add missing build args default once more

huggingface · Jan 23, 2025 · ba5cd39 · ba5cd39
1 parent 84fb23d
commit ba5cd39
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -108,7 +108,7 @@ jobs:
           labels: ${{ steps.meta.outputs.labels }}
           build-args: |
             PYTHON_VERSION: ${{ matrix.python-version }}
-            TENSORRT_LLM_VERSION: ${{ env.TENSORRT_LLM_VERSION }}
+            TRTLLM_VERSION: ${{ env.TENSORRT_LLM_VERSION }}
 
       - name: Generate artifact attestation
         uses: actions/attest-build-provenance@v2

diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,8 +1,7 @@
 ARG PYTHON_VERSION=3.10
-ARG TENSORRT_LLM_VERSION="latest"
+ARG TRTLLM_VERSION="latest"
 
-
-FROM tensorrt_llm/release:v${TENSORRT_LLM_VERSION}-py${PYTHON_VERSION}
+FROM tensorrt_llm/release:v${TRTLLM_VERSION}-py${PYTHON_VERSION}
 
 COPY . /opt/optimum-nvidia