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

.

Browse Source
This commit is contained in:
Nicolas Patry 2024-06-08 07:40:08 +02:00
parent fa3e811672
commit 909e6569d1
No known key found for this signature in database
GPG Key ID: 690EBED7128A9DC8
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
.github/workflows/build.yaml vendored
View File

@ -47,13 +47,15 @@ jobs:
export dockerfile="Dockerfile"
export label_extension=""
export docker_devices=""
export runs_on="[self-hosted, nvidia-gpu , multi-gpu, 4-a10, ci]"
# export runs_on="[self-hosted, nvidia-gpu , multi-gpu, 4-a10, ci]"
export runs_on="nvidia-gpu"
;;
rocm)
export dockerfile="Dockerfile_amd"
export label_extension="-rocm"
export docker_devices="/dev/kfd,/dev/dri"
export runs_on="[self-hosted, amd-gpu , multi-gpu]"
# export runs_on="[self-hosted, amd-gpu , multi-gpu]"
export runs_on="amd-gpu"
;;
intel)
export dockerfile="Dockerfile_intel"
@ -77,20 +79,20 @@ jobs:
authkey: ${{ secrets.TAILSCALE_AUTHKEY }}
- name: Login to GitHub Container Registry
if: github.event_name != 'pull_request'
uses: docker/login-action@v2
uses: docker/login-action@v3
with:
registry: ghcr.io
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}
- name: Login to internal Container Registry
uses: docker/login-action@v2.1.0
uses: docker/login-action@v3
with:
username: ${{ secrets.TAILSCALE_DOCKER_USERNAME }}
password: ${{ secrets.TAILSCALE_DOCKER_PASSWORD }}
registry: registry.internal.huggingface.tech
- name: Login to Azure Container Registry
if: github.event_name != 'pull_request'
uses: docker/login-action@v2.1.0
uses: docker/login-action@v3
with:
username: ${{ secrets.AZURE_DOCKER_USERNAME }}
password: ${{ secrets.AZURE_DOCKER_PASSWORD }}
@ -149,7 +151,8 @@ jobs:
group: ${{ github.workflow }}-${{ github.job }}-${{ needs.build-and-push.outputs.label }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: true
needs: build-and-push
runs-on: ${{ needs.build-and-push.outputs.runs_on }}
runs-on: ["self-hosted", "${{ needs.build-and-push.outputs.runs_on }}", "multi-gpu", "ci"]
if: needs.build-and-push.outputs.runs_on != 'ubuntu-latest'
steps:
- name: Checkout repository
uses: actions/checkout@v4