fix

huggingface · Jan 24, 2024 · 7b38326 · 7b38326
1 parent 565e081
commit 7b38326
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/3d_parallelism_unit_tests.yaml b/.github/workflows/3d_parallelism_unit_tests.yaml
@@ -1,4 +1,4 @@
-name: Run unit tests
+name: Run non-FA2-related unit tests
 
 on:
   workflow_dispatch:

diff --git a/.github/workflows/fa2_unit_tests.yaml b/.github/workflows/fa2_unit_tests.yaml
@@ -18,7 +18,7 @@ on:
 
 jobs:
   tests:
-    runs-on: [multi-gpu, nvidia-gpu, 8-t4, ci]
+    runs-on: [single-gpu, nvidia-gpu, a10, ci]
     container:
       image: runpod/pytorch:2.1.1-py3.10-cuda12.1.1-devel-ubuntu22.04
       ports: