diff --git a/.github/workflows/test-check.yaml b/.github/workflows/test-check.yaml index 362fd297321..c0709d4b01b 100644 --- a/.github/workflows/test-check.yaml +++ b/.github/workflows/test-check.yaml @@ -140,7 +140,7 @@ jobs: - name: "🔬 Running onnx tests" run: make test TARGETS=onnx pytorch-tests: - runs-on: aws-avx2-64G + runs-on: ubuntu-22.04 env: SPARSEZOO_TEST_MODE: "true" CLEARML_WEB_HOST: ${{ secrets.CLEARML_WEB_HOST }} @@ -169,7 +169,7 @@ jobs: - name: "🔬 Running pytorch tests" run: make test TARGETS=pytorch compat-pytorch-1_9-pytorch-tests: - runs-on: aws-avx2-64G + runs-on: ubuntu-22.04 env: SPARSEZOO_TEST_MODE: "true" CLEARML_WEB_HOST: ${{ secrets.CLEARML_WEB_HOST }} @@ -222,7 +222,7 @@ jobs: - name: "🔬 Running onnx tests" run: make test TARGETS=onnx transformers-tests: - runs-on: aws-avx2-64G + runs-on: ubuntu-22.04 env: SPARSEZOO_TEST_MODE: "true" CLEARML_WEB_HOST: ${{ secrets.CLEARML_WEB_HOST }}