From da528f6910c38fa829ba6f5921579d1eec3fb95d Mon Sep 17 00:00:00 2001
From: adtian2 <55163384+adtian2@users.noreply.github.com>
Date: Thu, 29 Aug 2024 14:57:27 -0700
Subject: [PATCH] SMP v2.5 (#4852)

* bumping smp version from 2.4 to 2.5

* removing old tsm-2.4 version

---------

Co-authored-by: Andrew Tian <tinandr@amazon.com>
---
 src/sagemaker/image_uri_config/pytorch-smp.json | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/sagemaker/image_uri_config/pytorch-smp.json b/src/sagemaker/image_uri_config/pytorch-smp.json
index ab1398666b..3a119c81d2 100644
--- a/src/sagemaker/image_uri_config/pytorch-smp.json
+++ b/src/sagemaker/image_uri_config/pytorch-smp.json
@@ -8,7 +8,7 @@
             "2.1": "2.1.2",
             "2.2": "2.3.1",
             "2.2.0": "2.3.1",
-            "2.3.1": "2.4.0"
+            "2.3.1": "2.5.0"
         },
         "versions": {
             "2.0.1": {
@@ -136,7 +136,7 @@
                 },
                 "repository": "smdistributed-modelparallel"
             },
-            "2.4.0": {
+            "2.5.0": {
                 "py_versions": [
                     "py311"
                 ],