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-07 19:46:34 +02:00
parent 87df3d5603
commit 5e769ce1e0
No known key found for this signature in database
GPG Key ID: 690EBED7128A9DC8
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -45,19 +45,20 @@ jobs:
dockerfile="Dockerfile"
label_extension=""
docker_devices=""
runs_on="[self-hosted, nvidia-gpu , multi-gpu, 4-a10, ci]"
# runs_on="[self-hosted, nvidia-gpu , multi-gpu, 4-a10, ci]"
runs_on="ubuntu-latest"
;;
rocm)
dockerfile="Dockerfile_amd"
label_extension="-rocm"
docker_devices="/dev/kfd,/dev/dri"
runs_on="[self-hosted, amd-gpu , multi-gpu]"
runs_on="ubuntu-latest"
;;
rocm)
dockerfile="Dockerfile_intel"
label_extension="-intel"
docker_devices=""
runs_on=""
runs_on="ubuntu-latest"
;;
esac
echo "DOCKERFILE=${dockerfile}" >> $GITHUB_ENV