fix: Correctly tag docker images

huggingface · Jan 6, 2025 · 2042149 · 2042149
1 parent b4eaaff
commit 2042149
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/.github/workflows/build_trtllm.yaml b/.github/workflows/build_trtllm.yaml
@@ -30,11 +30,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 +102,6 @@ jobs:
 
       - name: Build and push Docker image
         id: build-and-push
-        env:
-          LABEL: "trtllm"
         uses: docker/build-push-action@v6
         with:
           context: .