change runner cluster for A10

huggingface · Sep 26, 2024 · 565b3a4 · 565b3a4
1 parent 97c13b0
commit 565b3a4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/fa2_unit_tests.yaml b/.github/workflows/fa2_unit_tests.yaml
@@ -19,7 +19,8 @@ on:
 
 jobs:
   tests:
-    runs-on: [single-gpu, nvidia-gpu, a10, ci]
+    runs-on:
+      group: aws-g5-4xlarge-plus
     container:
       image: runpod/pytorch:2.1.1-py3.10-cuda12.1.1-devel-ubuntu22.04
       ports: