From a3015590430c67c51267612e79ee8d43212448dd Mon Sep 17 00:00:00 2001
From: Hugo Larcher <hugo.larcher@huggingface.co>
Date: Mon, 6 Jan 2025 12:09:56 +0100
Subject: [PATCH] fix: Correctly tag docker images (#2878)

* fix: Correctly tag docker images

* fix: Correctly tag docker images
---
 .github/workflows/build_trtllm.yaml | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/build_trtllm.yaml b/.github/workflows/build_trtllm.yaml
index 054b5d89..1f1b8784 100644
--- a/.github/workflows/build_trtllm.yaml
+++ b/.github/workflows/build_trtllm.yaml
@@ -16,6 +16,7 @@ on:
       - "launcher/**"
       - "Cargo.lock"
       - "rust-toolchain.toml"
+      - ".github/workflows/build_trtllm.yaml"
     branches:
       - "main"
 
@@ -30,11 +31,18 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
+      - name: Inject slug/short variables
+        uses: rlespinasse/github-slug-action@v4.4.1
+
       - name: Extract TensorRT-LLM version
         run: |
           echo "TENSORRT_LLM_VERSION=$(grep -oP '([a-z,0-9]{40})' $GITHUB_WORKSPACE/backends/trtllm/cmake/trtllm.cmake)" >> $GITHUB_ENV
           echo "TensorRT-LLM version: ${{ env.TENSORRT_LLM_VERSION }}"
 
+      - name: Set LABEL
+        run: |
+          echo "LABEL=trtllm" >> $GITHUB_ENV
+
       - name: "Configure AWS Credentials"
         id: aws-creds
         uses: aws-actions/configure-aws-credentials@v4
@@ -95,8 +103,6 @@ jobs:
 
       - name: Build and push Docker image
         id: build-and-push
-        env:
-          LABEL: "trtllm"
         uses: docker/build-push-action@v6
         with:
           context: .