mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-06-09 02:42:05 +00:00
.
This commit is contained in:
parent
0ced5fac2d
commit
eec6c3241b
4
.github/workflows/build.yaml
vendored
4
.github/workflows/build.yaml
vendored
@ -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
|
||||||
|
Loading…
Reference in New Issue
Block a user