huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 20:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

?

Browse Source
This commit is contained in:
Nicolas Patry 2024-10-01 15:57:20 +02:00
parent 290bd2ec4c
commit f2b1429b9b
No known key found for this signature in database
GPG Key ID: E939E8CC91A1C674
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -41,7 +41,6 @@ jobs:
- name: Construct harware variables
shell: bash
run: |
export extra_pytest = ""
case ${{ inputs.hardware }} in
cuda)
export dockerfile="Dockerfile"
@ -49,6 +48,7 @@ jobs:
export docker_devices=""
export runs_on="aws-g6-12xl-plus-priv-cache"
export platform=""
export extra_pytest = " "
;;
rocm)
export dockerfile="Dockerfile_amd"
@ -64,6 +64,7 @@ jobs:
export docker_devices=""
export runs_on="ubuntu-latest"
export platform="xpu"
export extra_pytest = " "
;;
intel-cpu)
export dockerfile="Dockerfile_intel"
@ -71,6 +72,7 @@ jobs:
export docker_devices=""
export runs_on="ubuntu-latest"
export platform="cpu"
export extra_pytest = " "
;;
esac
echo $dockerfile