diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 0df645ca8..c5d7370e0 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -52,7 +52,7 @@ 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="amd-gpu-runners" export extra_pytest="-k flash_llama_grammar_tools" ;; intel)