From db3259a03523ff87aed94c610c644d04ab7787b8 Mon Sep 17 00:00:00 2001
From: Andrew Tian <tinandr@amazon.com>
Date: Tue, 29 Oct 2024 15:21:24 -0700
Subject: [PATCH] bumping smp version from 2.6.0 to 2.6.1

---
 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 18eb0fb4c3..91d8ab9184 100644
--- a/src/sagemaker/image_uri_config/pytorch-smp.json
+++ b/src/sagemaker/image_uri_config/pytorch-smp.json
@@ -9,7 +9,7 @@
             "2.2": "2.3.1",
             "2.2.0": "2.3.1",
             "2.3.1": "2.5.0",
-            "2.4.1": "2.6.0"
+            "2.4.1": "2.6.1"
         },
         "versions": {
             "2.0.1": {
@@ -162,7 +162,7 @@
                 },
                 "repository": "smdistributed-modelparallel"
             },
-            "2.6.0": {
+            "2.6.1": {
                 "py_versions": [
                     "py311"
                 ],