huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 12:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Correctly tag docker images (#2878)

Browse Source 
* fix: Correctly tag docker images

* fix: Correctly tag docker images
This commit is contained in:
Hugo Larcher 2025-01-06 12:09:56 +01:00 committed by Morgan Funtowicz
parent 7c838856d9
commit a301559043
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
.github/workflows/build_trtllm.yaml vendored
View File

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