From ba5cd3944301399161d788a941d3086690610d54 Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Thu, 23 Jan 2025 12:56:39 +0100
Subject: [PATCH] misc(gha): add missing build args default once more

---
 .github/workflows/release.yml | 2 +-
 docker/Dockerfile             | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 81a1a56f..95ad4bc6 100644
--- 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
index 0891e958..4ef077ff 100644
--- 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