change: Add image configs and region config for TPE (ap-east-2) (#5167)

* add image configs and region config for TPE (ap-east-2)

* remove TPE from djl-neuronx

---------

Co-authored-by: isha chidrawar <[email protected]>
Co-authored-by: pintaoz-aws <[email protected]>

Author: 3 people

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",

src/sagemaker/image_uri_config/huggingface-neuronx.json

@@ -21,6 +21,7 @@
                     ],
                     "repository": "huggingface-pytorch-training-neuronx",
                     "registries": {
+                        "ap-east-2": "975050140332",
                         "ap-northeast-1": "763104351884",
                         "ap-south-1": "763104351884",
                         "ap-south-2": "772153158452",
@@ -64,6 +65,7 @@
                     ],
                     "repository": "huggingface-pytorch-inference-neuronx",
                     "registries": {
+                        "ap-east-2": "975050140332",
                         "ap-northeast-1": "763104351884",
                         "ap-south-1": "763104351884",
                         "ap-south-2": "772153158452",
@@ -107,6 +109,7 @@
                     ],
                     "repository": "huggingface-pytorch-inference-neuronx",
                     "registries": {
+                        "ap-east-2": "975050140332",
                         "ap-northeast-1": "763104351884",
                         "ap-south-1": "763104351884",
                         "ap-south-2": "772153158452",
@@ -252,6 +255,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",
@@ -307,6 +311,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",
@@ -363,6 +368,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",
@@ -413,6 +419,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",

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",

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",