Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

.

This commit is contained in:
Nicolas Patry 2024-06-08 21:55:27 +02:00
parent 0ced5fac2d
commit eec6c3241b
No known key found for this signature in database
GPG Key ID: 690EBED7128A9DC8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -48,15 +48,14 @@ jobs:
export dockerfile="Dockerfile" export dockerfile="Dockerfile"
export label_extension="" export label_extension=""
export docker_devices="" export docker_devices=""
# export runs_on="[self-hosted, nvidia-gpu , multi-gpu, 4-a10, ci]"
export runs_on="nvidia-gpu" export runs_on="nvidia-gpu"
;; ;;
rocm) rocm)
export dockerfile="Dockerfile_amd" export dockerfile="Dockerfile_amd"
export label_extension="-rocm" export label_extension="-rocm"
export docker_devices="/dev/kfd,/dev/dri" export docker_devices="/dev/kfd,/dev/dri"
# export runs_on="[self-hosted, amd-gpu , multi-gpu]"
export runs_on="amd-gpu-tgi" export runs_on="amd-gpu-tgi"
# export runs_on="ubuntu-latest"
;; ;;
intel) intel)
export dockerfile="Dockerfile_intel" export dockerfile="Dockerfile_intel"
@ -168,6 +167,7 @@ jobs:
make install-integration-tests make install-integration-tests
- name: Tailscale - name: Tailscale
uses: huggingface/tailscale-action@main uses: huggingface/tailscale-action@main
if: needs.build-and-push.outputs.runs_on != 'amd-gpu-tgi'
with: with:
authkey: ${{ secrets.TAILSCALE_AUTHKEY }} authkey: ${{ secrets.TAILSCALE_AUTHKEY }}
- name: Run tests - name: Run tests