Update default SM local gpu instance type (#4443)

aws · Nov 22, 2024 · 13ee94f · 13ee94f
1 parent d86c644
commit 13ee94f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_utils/sagemaker.py b/test/test_utils/sagemaker.py
@@ -87,7 +87,7 @@ def assign_sagemaker_local_job_instance_type(image):
         return "p3.2xlarge"
     elif all(word in image for word in ["huggingface-pytorch", "training", "gpu"]):
         return "g5.8xlarge"
-    return "p4d.24xlarge" if "gpu" in image else "c5.18xlarge"
+    return "g5.12xlarge" if "gpu" in image else "c5.18xlarge"
 
 
 def assign_sagemaker_local_test_ami(image, region, instance_type):