Fix hanging pipeline fetching (#7837)

update
huggingface · May 2, 2024 · 3ffa7b4 · 3ffa7b4
1 parent c1b2a89
commit 3ffa7b4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.github/workflows/push_tests.yml b/.github/workflows/push_tests.yml
@@ -21,7 +21,9 @@ env:
 jobs:
   setup_torch_cuda_pipeline_matrix:
     name: Setup Torch Pipelines CUDA Slow Tests Matrix
-    runs-on: diffusers/diffusers-pytorch-cpu
+    runs-on: [ self-hosted, intel-cpu, 8-cpu, ci ]
+    container:
+      image: diffusers/diffusers-pytorch-cpu
     outputs:
       pipeline_test_matrix: ${{ steps.fetch_pipeline_matrix.outputs.pipeline_test_matrix }}
     steps: