From f5cdb291d76f993fab84a479d723de7f17915c3f Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Mon, 6 Jan 2025 22:59:19 +0100
Subject: [PATCH] misc(ci): go

---
 .github/workflows/build_trtllm.yaml | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/build_trtllm.yaml b/.github/workflows/build_trtllm.yaml
index 603d399988c..2928f6cb28a 100644
--- a/.github/workflows/build_trtllm.yaml
+++ b/.github/workflows/build_trtllm.yaml
@@ -30,7 +30,7 @@ jobs:
       docker_image: ${{ steps.final.outputs.docker_image }}
       label: ${ steps.final.outputs.label }}
     concurrency:
-      group: ${{ github.workflow }}-${{ github.job }}-trtllm-${{ github.head_ref || github.run_id }}
+      group: ${{ github.workflow }}-${{ github.job }}-build-and-push-${{ github.head_ref || github.run_id }}
       cancel-in-progress: true
     runs-on:
       group: aws-highmemory-64-plus-priv
@@ -141,10 +141,8 @@ jobs:
       cancel-in-progress: true
     runs-on:
       group: aws-g6-12xl-plus-priv-cache
-    env:
-      DOCKER_IMAGE: ${{ needs.build-and-push.outputs.docker_image }}
     container:
-      image: ${DOCKER_IMAGE}
+      image: ${{ needs.build-and-push.outputs.docker_image }}
       credentials:
         username: ${{ secrets.REGISTRY_USERNAME }}
         password: ${{ secrets.REGISTRY_PASSWORD }}