huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-22 23:42:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

?

Browse Source
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" dockerfile="Dockerfile"
label_extension="" label_extension=""
docker_devices="" 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) rocm)
dockerfile="Dockerfile_amd" dockerfile="Dockerfile_amd"
label_extension="-rocm" label_extension="-rocm"
docker_devices="/dev/kfd,/dev/dri" docker_devices="/dev/kfd,/dev/dri"
runs_on="[self-hosted, amd-gpu , multi-gpu]" runs_on="ubuntu-latest"
;; ;;
rocm) rocm)
dockerfile="Dockerfile_intel" dockerfile="Dockerfile_intel"
label_extension="-intel" label_extension="-intel"
docker_devices="" docker_devices=""
runs_on="" runs_on="ubuntu-latest"
;; ;;
esac esac
echo "DOCKERFILE=${dockerfile}" >> $GITHUB_ENV echo "DOCKERFILE=${dockerfile}" >> $GITHUB_ENV