move CI/Cuda runner

huggingface · Sep 26, 2024 · a0184a4 · a0184a4
1 parent 10d4718
commit a0184a4
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