fix: Correctly tag docker images (#2878)

* fix: Correctly tag docker images * fix: Correctly tag docker images
huggingface · Jan 6, 2025 · 7180c64 · 7180c64
1 parent b4eaaff
commit 7180c64
Showing 1 changed file with 8 additions and 2 deletions.
diff --git 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/[email protected]
+
       - 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: .