misc(llamacpp): maybe?

huggingface · Jan 6, 2025 · 764315d · 764315d
1 parent b8084ea
commit 764315d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/.github/workflows/build_trtllm.yaml b/.github/workflows/build_trtllm.yaml
@@ -128,11 +128,9 @@ jobs:
     needs: build
     runs-on:
       group: aws-g6-12xl-plus-priv-cache
-    env:
-      TARGET_SHA: ${{ env.LABEL }}
     container:
       image:
-        registry.internal.huggingface.tech/api-inference/community/text-generation-inference/tensorrt-llm:${TARGET_SHA}
+        registry.internal.huggingface.tech/api-inference/community/text-generation-inference/tensorrt-llm:${{ github.GITHUB_SHA }}
       credentials:
         username: ${{ secrets.REGISTRY_USERNAME }}
         password: ${{ secrets.REGISTRY_PASSWORD }}