diff --git a/.github/workflows/push-important-models.yml b/.github/workflows/push-important-models.yml index a013dbc524c..5a6aeec2bd6 100644 --- a/.github/workflows/push-important-models.yml +++ b/.github/workflows/push-important-models.yml @@ -53,7 +53,6 @@ jobs: test_modified_files: needs: get_modified_models name: Slow & FA2 tests - runs-on: ubuntu-latest runs-on: [single-gpu, nvidia-gpu, a10, ci] container: image: huggingface/transformers-all-latest-gpu