use available

huggingface · Sep 28, 2024 · 3eecee6 · 3eecee6
1 parent 806faca
commit 3eecee6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/test_onnxruntime_gpu.yml b/.github/workflows/test_onnxruntime_gpu.yml
@@ -29,7 +29,7 @@ jobs:
       }}
 
     runs-on:
-      group: aws-g5-4xlarge-plus
+      group: aws-g6-4xlarge-plus
 
     container:
       image: nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04

diff --git a/.github/workflows/test_onnxruntime_training.yml b/.github/workflows/test_onnxruntime_training.yml
@@ -29,7 +29,7 @@ jobs:
       }}
 
     runs-on:
-      group: aws-g5-4xlarge-plus
+      group: aws-g6-4xlarge-plus
 
     container:
       image: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04