mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-26 12:32:10 +00:00
.
This commit is contained in:
parent
eec6c3241b
commit
41699e9bbf
5
.github/workflows/build.yaml
vendored
5
.github/workflows/build.yaml
vendored
@ -54,8 +54,9 @@ jobs:
|
||||
export dockerfile="Dockerfile_amd"
|
||||
export label_extension="-rocm"
|
||||
export docker_devices="/dev/kfd,/dev/dri"
|
||||
export runs_on="amd-gpu-tgi"
|
||||
# export runs_on="ubuntu-latest"
|
||||
# TODO Re-enable when they pass.
|
||||
# export runs_on="amd-gpu-tgi"
|
||||
export runs_on="ubuntu-latest"
|
||||
;;
|
||||
intel)
|
||||
export dockerfile="Dockerfile_intel"
|
||||
|
Loading…
Reference in New Issue
Block a user