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

misc(ci): gogogo

Browse Source
This commit is contained in:
Morgan Funtowicz 2025-01-06 15:23:11 +01:00
parent 51aefb15d0
commit 0c972cdd28
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -126,20 +126,16 @@ jobs:
- name: Final - name: Final
id: final id: final
run: | run: |
echo "docker_image=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-${{ env.GITHUB_SHA_SHORT}}${{ env.LABEL }}" >> "$GITHUB_OUTPUT" echo "docker_image=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-${{ env.GITHUB_SHA_SHORT }}${{ env.LABEL }}" >> "$GITHUB_OUTPUT"
echo "docker_devices=${{ env.DOCKER_DEVICES }}" >> "$GITHUB_OUTPUT"
echo "docker_volume=${{ env.DOCKER_VOLUME }}" >> "$GITHUB_OUTPUT"
echo "runs_on=${{ env.RUNS_ON }}" >> "$GITHUB_OUTPUT"
echo "label=${{ env.LABEL }}" >> "$GITHUB_OUTPUT"
echo "extra_pytest=${{ env.EXTRA_PYTEST }}" >> "$GITHUB_OUTPUT"
tests: tests:
needs: build-and-push needs: build-and-push
runs-on: runs-on:
group: aws-g6-12xl-plus-priv-cache group: aws-g6-12xl-plus-priv-cache
env:
DOCKER_IMAGE: ${{ needs.build-and-push.outputs.docker_image }}
container: container:
image: image: ${DOCKER_IMAGE}
registry.internal.huggingface.tech/api-inference/community/text-generation-inference/tensorrt-llm:${{ needs.build-and-push.outputs.docker_image }}
credentials: credentials:
username: ${{ secrets.REGISTRY_USERNAME }} username: ${{ secrets.REGISTRY_USERNAME }}
password: ${{ secrets.REGISTRY_PASSWORD }} password: ${{ secrets.REGISTRY_PASSWORD }}