From b20b5914a2f123f0245cc799f15cf18196d076c8 Mon Sep 17 00:00:00 2001 From: isha chidrawar Date: Fri, 9 May 2025 15:54:14 -0400 Subject: [PATCH 1/2] add image configs and region config for TPE (ap-east-2) --- src/sagemaker/image_uri_config/djl-neuronx.json | 8 ++++++++ src/sagemaker/image_uri_config/huggingface-neuron.json | 1 + src/sagemaker/image_uri_config/huggingface-neuronx.json | 7 +++++++ .../image_uri_config/huggingface-training-compiler.json | 3 +++ src/sagemaker/jumpstart/region_config.json | 4 ++++ 5 files changed, 23 insertions(+) diff --git a/src/sagemaker/image_uri_config/djl-neuronx.json b/src/sagemaker/image_uri_config/djl-neuronx.json index 1fd7492ff4..79e16b97b8 100644 --- a/src/sagemaker/image_uri_config/djl-neuronx.json +++ b/src/sagemaker/image_uri_config/djl-neuronx.json @@ -8,6 +8,7 @@ "versions": { "0.29.0": { "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -34,6 +35,7 @@ }, "0.28.0": { "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -60,6 +62,7 @@ }, "0.27.0": { "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -86,6 +89,7 @@ }, "0.26.0": { "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -112,6 +116,7 @@ }, "0.25.0": { "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -138,6 +143,7 @@ }, "0.24.0": { "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -164,6 +170,7 @@ }, "0.23.0": { "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -190,6 +197,7 @@ }, "0.22.1": { "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", diff --git a/src/sagemaker/image_uri_config/huggingface-neuron.json b/src/sagemaker/image_uri_config/huggingface-neuron.json index 4e950bdb70..2a68282327 100644 --- a/src/sagemaker/image_uri_config/huggingface-neuron.json +++ b/src/sagemaker/image_uri_config/huggingface-neuron.json @@ -17,6 +17,7 @@ ], "repository": "huggingface-pytorch-inference-neuron", "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-south-2": "772153158452", diff --git a/src/sagemaker/image_uri_config/huggingface-neuronx.json b/src/sagemaker/image_uri_config/huggingface-neuronx.json index a3426d5e0c..080d4806fa 100644 --- a/src/sagemaker/image_uri_config/huggingface-neuronx.json +++ b/src/sagemaker/image_uri_config/huggingface-neuronx.json @@ -19,6 +19,7 @@ ], "repository": "huggingface-pytorch-training-neuronx", "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-south-2": "772153158452", @@ -62,6 +63,7 @@ ], "repository": "huggingface-pytorch-inference-neuronx", "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-south-2": "772153158452", @@ -105,6 +107,7 @@ ], "repository": "huggingface-pytorch-inference-neuronx", "registries": { + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-south-2": "772153158452", @@ -163,6 +166,7 @@ "af-south-1": "626614931356", "il-central-1": "780543022126", "ap-east-1": "871362719292", + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-northeast-2": "763104351884", "ap-northeast-3": "364406365360", @@ -218,6 +222,7 @@ "af-south-1": "626614931356", "il-central-1": "780543022126", "ap-east-1": "871362719292", + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-northeast-2": "763104351884", "ap-northeast-3": "364406365360", @@ -274,6 +279,7 @@ "af-south-1": "626614931356", "il-central-1": "780543022126", "ap-east-1": "871362719292", + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-northeast-2": "763104351884", "ap-northeast-3": "364406365360", @@ -324,6 +330,7 @@ "af-south-1": "626614931356", "il-central-1": "780543022126", "ap-east-1": "871362719292", + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-northeast-2": "763104351884", "ap-northeast-3": "364406365360", diff --git a/src/sagemaker/image_uri_config/huggingface-training-compiler.json b/src/sagemaker/image_uri_config/huggingface-training-compiler.json index fa3a4119ca..c84469acc2 100644 --- a/src/sagemaker/image_uri_config/huggingface-training-compiler.json +++ b/src/sagemaker/image_uri_config/huggingface-training-compiler.json @@ -60,6 +60,7 @@ "af-south-1": "626614931356", "il-central-1": "780543022126", "ap-east-1": "871362719292", + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-northeast-2": "763104351884", "ap-northeast-3": "364406365360", @@ -103,6 +104,7 @@ "af-south-1": "626614931356", "il-central-1": "780543022126", "ap-east-1": "871362719292", + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-northeast-2": "763104351884", "ap-northeast-3": "364406365360", @@ -151,6 +153,7 @@ "af-south-1": "626614931356", "il-central-1": "780543022126", "ap-east-1": "871362719292", + "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-northeast-2": "763104351884", "ap-northeast-3": "364406365360", diff --git a/src/sagemaker/jumpstart/region_config.json b/src/sagemaker/jumpstart/region_config.json index 30bea6ee70..136bf8256c 100644 --- a/src/sagemaker/jumpstart/region_config.json +++ b/src/sagemaker/jumpstart/region_config.json @@ -7,6 +7,10 @@ "content_bucket": "jumpstart-cache-prod-ap-east-1", "gated_content_bucket": "jumpstart-private-cache-prod-ap-east-1" }, + "ap-east-2": { + "content_bucket": "jumpstart-cache-prod-ap-east-2", + "gated_content_bucket": "jumpstart-private-cache-prod-ap-east-2" + }, "ap-northeast-1": { "content_bucket": "jumpstart-cache-prod-ap-northeast-1", "gated_content_bucket": "jumpstart-private-cache-prod-ap-northeast-1", From 05d8b20ea16422913cee2ee01118d27cabcd7a67 Mon Sep 17 00:00:00 2001 From: isha chidrawar Date: Mon, 12 May 2025 15:17:27 -0400 Subject: [PATCH 2/2] remove TPE from djl-neuronx --- src/sagemaker/image_uri_config/djl-neuronx.json | 8 -------- 1 file changed, 8 deletions(-) diff --git a/src/sagemaker/image_uri_config/djl-neuronx.json b/src/sagemaker/image_uri_config/djl-neuronx.json index 79e16b97b8..1fd7492ff4 100644 --- a/src/sagemaker/image_uri_config/djl-neuronx.json +++ b/src/sagemaker/image_uri_config/djl-neuronx.json @@ -8,7 +8,6 @@ "versions": { "0.29.0": { "registries": { - "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -35,7 +34,6 @@ }, "0.28.0": { "registries": { - "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -62,7 +60,6 @@ }, "0.27.0": { "registries": { - "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -89,7 +86,6 @@ }, "0.26.0": { "registries": { - "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -116,7 +112,6 @@ }, "0.25.0": { "registries": { - "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -143,7 +138,6 @@ }, "0.24.0": { "registries": { - "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -170,7 +164,6 @@ }, "0.23.0": { "registries": { - "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884", @@ -197,7 +190,6 @@ }, "0.22.1": { "registries": { - "ap-east-2": "975050140332", "ap-northeast-1": "763104351884", "ap-south-1": "763104351884", "ap-southeast-1": "763104351884",