Merge branch 'master' into autopatch

aws · Nov 22, 2024 · 203a49b · 203a49b
2 parents ff23feb + 2d62185
commit 203a49b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/release_images_inference.yml b/release_images_inference.yml
@@ -68,7 +68,7 @@ release_images:
       cuda_version: "cu124"
       example: False
       disable_sm_tag: True
-      force_release: True
+      force_release: False
   6:
     framework: "huggingface_pytorch"
     version: "2.1.0"

diff --git a/release_images_training.yml b/release_images_training.yml
@@ -55,12 +55,12 @@ release_images:
     framework: "huggingface_pytorch"
     version: "2.1.2"
     arch_type: "x86"
-    hf_transformers: "4.41.1"
+    hf_transformers: "4.43.2"
     training:
       device_types: [ "neuronx" ]
       python_versions: [ "py310" ]
       os_version: "ubuntu20.04"
-      neuron_sdk_version: "sdk2.19.1"
+      neuron_sdk_version: "sdk2.20.0"
       example: False
       disable_sm_tag: True
       force_release: False

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):