Merge pull request huggingface#2507 from huggingface/ci-move

move CI/Cuda runner
getong · Sep 26, 2024 · c3c392f · c3c392f
2 parents 10d4718 + a0184a4
commit c3c392f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/ci_cuda.yaml b/.github/workflows/ci_cuda.yaml
@@ -9,7 +9,8 @@ jobs:
     concurrency:
       group: ${{ github.workflow }}-${{ github.job }}-${{ github.head_ref || github.run_id }}
       cancel-in-progress: true
-    runs-on: [single-gpu, nvidia-gpu, t4, ci]
+    runs-on:
+      group: aws-g4dn-2xlarge
     container:
       image: nvidia/cuda:12.3.1-devel-ubuntu22.04
       options: --gpus 0