update

Author: msaroufim

.github/workflows/runner_ci.yml

@@ -12,7 +12,7 @@ on:
 
 jobs:
   check-cuda:
-    runs-on: gpu-t4
+    runs-on: gpu-t4-4-core
     timeout-minutes: 10
     container:
       image: nvidia/cuda:12.4.0-devel-ubuntu22.04
@@ -48,7 +48,7 @@ jobs:
       CUDA_VISIBLE_DEVICES: 0
 
   check-pytorch:
-    runs-on: gpu-t4
+    runs-on: gpu-t4-4-core
     timeout-minutes: 10
     container:
       image: nvidia/cuda:12.4.0-devel-ubuntu22.04