chore: minor

flexflow · Jan 15, 2025 · 13c29af · 13c29af
1 parent 6a45d77
commit 13c29af
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.github/workflows/gpu-ci.yml b/.github/workflows/gpu-ci.yml
@@ -7,7 +7,10 @@ concurrency:
 jobs:
   gpu-ci:
     name: GPU-CI
-    runs-on: [runs-on,runner=gpu-nvidia-lightweight]
+    runs-on:
+      - runs-on
+      - family=g4dn.xlarge
+      - image=ubuntu22-full-x64
 
     strategy:
       max-parallel: 1