From ff9a03e05245fb869118c7a2c9ee4ecf8b6150e6 Mon Sep 17 00:00:00 2001
From: Mark Bunday <mabunday@amazon.com>
Date: Fri, 22 Mar 2024 14:57:35 -0400
Subject: [PATCH 1/2] Fix build dependencies

---
 .../pytorch/training/docker/1.13/py3/cu117/Dockerfile.gpu    | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/huggingface/pytorch/training/docker/1.13/py3/cu117/Dockerfile.gpu b/huggingface/pytorch/training/docker/1.13/py3/cu117/Dockerfile.gpu
index 4c0b8f98ced0..840a8052b72e 100644
--- a/huggingface/pytorch/training/docker/1.13/py3/cu117/Dockerfile.gpu
+++ b/huggingface/pytorch/training/docker/1.13/py3/cu117/Dockerfile.gpu
@@ -20,8 +20,9 @@ RUN pip install --no-cache-dir \
     transformers[sklearn,sentencepiece,audio,vision]==${TRANSFORMERS_VERSION} \
     datasets==${DATASETS_VERSION} \
     $PT_TORCHAUDIO_URL \
-    multiprocess~=0.70.15 \
-    dill~=0.3.7
+    multiprocess==0.70.14 \
+    dill==0.3.6 \
+    sagemaker==2.132.0
 
 RUN apt-get update \
  # TODO: Remove upgrade statements once packages are updated in base image

From cb9aebf3d83babdedd8045a3b728c8d071eb5e9a Mon Sep 17 00:00:00 2001
From: Mark Bunday <mabunday@amazon.com>
Date: Fri, 22 Mar 2024 15:42:23 -0400
Subject: [PATCH 2/2] Release neuronx

---
 release_images_training.yml | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/release_images_training.yml b/release_images_training.yml
index b209720aa9b2..c383aa85de8f 100644
--- a/release_images_training.yml
+++ b/release_images_training.yml
@@ -78,3 +78,16 @@ release_images:
       example: False
       disable_sm_tag: False
       force_release: False
+  7:
+    framework: "huggingface_pytorch"
+    version: "1.13.1"
+    customer_type: "sagemaker"
+    arch_type: "x86"
+    training:
+      device_types: [ "neuronx" ]
+      python_versions: [ "py310" ]
+      os_version: "ubuntu20.04"
+      neuron_sdk_version: "sdk2.15.0"
+      example: False
+      disable_sm_tag: False
+      force_release: False
\ No newline at end of file