diff --git a/.github/workflows/build_and_test.yml b/.github/workflows/build_and_test.yml index c78504c1e..b03430c89 100644 --- a/.github/workflows/build_and_test.yml +++ b/.github/workflows/build_and_test.yml @@ -128,7 +128,7 @@ jobs: fi chatgpt_api_tests: - runs-on: ${{ (matrix.inference_engine == 'tinygrad' || matrix.inference_engine == 'dummy') && 'depot-ubuntu-22.04-4' || 'depot-macos-latest' }} + runs-on: ${{ (matrix.inference_engine == 'tinygrad' || matrix.inference_engine == 'dummy') && 'M4PRO_GPU16_24GB' || 'M4PRO_GPU16_24GB' }} strategy: matrix: inference_engine: [mlx, tinygrad, dummy]