test cuda

pytorch · Apr 17, 2024 · dbcc91f · dbcc91f
1 parent d8c30a3
commit dbcc91f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.ci/scripts/gather_test_models.py b/.ci/scripts/gather_test_models.py
@@ -28,7 +28,7 @@
         "macos-14": "aarch64",
     },
     "gpu": {
-        "linux.g5.4xlarge.nvidia.gpu": "cuda",
+        "linux.g5.12xlarge.nvidia.gpu": "cuda",
     },
 }