diff --git a/.github/workflows/vllm_benchmark.yaml b/.github/workflows/vllm_benchmark.yaml index f0c7c2230c..308fea10d2 100644 --- a/.github/workflows/vllm_benchmark.yaml +++ b/.github/workflows/vllm_benchmark.yaml @@ -94,7 +94,7 @@ jobs: vllm_benchmark: if: ${{ !github.event.pull_request.head.repo.fork }} - runs-on: aiter-8gpu-runner + runs-on: aiter-mi325-8gpu-bm needs: build_vllm_image strategy: fail-fast: false