From 4f87837ed627744714cdce8662ac4306e5966074 Mon Sep 17 00:00:00 2001
From: Morgan Funtowicz <funtowiczmo@gmail.com>
Date: Mon, 6 Jan 2025 14:13:24 +0100
Subject: [PATCH] misc(llamacpp): maybe?

---
 .github/workflows/build_trtllm.yaml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/build_trtllm.yaml b/.github/workflows/build_trtllm.yaml
index 1f1b8784..788f8539 100644
--- a/.github/workflows/build_trtllm.yaml
+++ b/.github/workflows/build_trtllm.yaml
@@ -128,9 +128,11 @@ 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:${{ env.LABEL }}
+        registry.internal.huggingface.tech/api-inference/community/text-generation-inference/tensorrt-llm:${TARGET_SHA}
       credentials:
         username: ${{ secrets.REGISTRY_USERNAME }}
         password: ${{ secrets.REGISTRY_PASSWORD }}