From 1592b6fec374e4355e58fbc4579b6abb01834019 Mon Sep 17 00:00:00 2001
From: malavhs <malavhs@amazon.com>
Date: Thu, 24 Oct 2024 21:24:45 +0000
Subject: [PATCH] Change the instance type

---
 test/test_utils/sagemaker.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/test_utils/sagemaker.py b/test/test_utils/sagemaker.py
index be6c6e483360..4026c0a51a0c 100644
--- a/test/test_utils/sagemaker.py
+++ b/test/test_utils/sagemaker.py
@@ -86,7 +86,7 @@ def assign_sagemaker_local_job_instance_type(image):
     elif "trcomp" in image:
         return "p3.2xlarge"
     elif all(word in image for word in ["huggingface-pytorch", "training", "gpu"]):
-        return "g5.8xlarge"
+        return "p5.48xlarge"
     return "p3.8xlarge" if "gpu" in image else "c5.18xlarge"