[Diffusers CI] Switch slow test runners (#7123)

update

[Diffusers CI] Switch slow test runners (#7123)
update
e6cf2190 · Dhruv Nair · GitHub · 7db935a1 · e6cf2190
Unverified Commit e6cf2190 authored Feb 28, 2024 by Dhruv Nair Committed by GitHub Feb 28, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

.github/workflows/push_tests.yml .github/workflows/push_tests.yml +1 -1

No files found.
--- a/.github/workflows/push_tests.yml
+++ b/.github/workflows/push_tests.yml
@@ -64,7 +64,7 @@ jobs:
      max-parallel: 1
      matrix:
        module: ${{ fromJson(needs.setup_torch_cuda_pipeline_matrix.outputs.pipeline_test_matrix) }}
-    runs-on: docker-gpu
+    runs-on: [single-gpu, nvidia-gpu, t4, ci]
    container:
      image: diffusers/diffusers-pytorch-cuda
      options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ --gpus 0