Skip to content

Commit 391be37

Browse files
tlian25txlianbenieric
authored
[change] add us-gov and cn region repo accounts to djl and hugging face image metadata (#4863)
* add us-gov region repo accounts to djl image metadata * add us-gov and cn region repo accounts to djl and hugging face image metadata * change: update china region jumpstart buckets --------- Co-authored-by: txlian <txlian@amazon.com> Co-authored-by: Erick Benitez-Ramos <141277478+benieric@users.noreply.github.com>
1 parent adb13e8 commit 391be37

File tree

6 files changed

+82
-0
lines changed

6 files changed

+82
-0
lines changed

‎src/sagemaker/image_uri_config/djl-neuronx.json‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@
99
"ap-south-1": "763104351884",
1010
"ap-southeast-1": "763104351884",
1111
"ap-southeast-2": "763104351884",
12+
"cn-north-1": "727897471807",
13+
"cn-northwest-1": "727897471807",
1214
"eu-central-1": "763104351884",
1315
"eu-west-1": "763104351884",
1416
"eu-west-3": "763104351884",
@@ -29,6 +31,8 @@
2931
"ap-south-1": "763104351884",
3032
"ap-southeast-1": "763104351884",
3133
"ap-southeast-2": "763104351884",
34+
"cn-north-1": "727897471807",
35+
"cn-northwest-1": "727897471807",
3236
"eu-central-1": "763104351884",
3337
"eu-west-1": "763104351884",
3438
"eu-west-3": "763104351884",
@@ -49,6 +53,8 @@
4953
"ap-south-1": "763104351884",
5054
"ap-southeast-1": "763104351884",
5155
"ap-southeast-2": "763104351884",
56+
"cn-north-1": "727897471807",
57+
"cn-northwest-1": "727897471807",
5258
"eu-central-1": "763104351884",
5359
"eu-west-1": "763104351884",
5460
"eu-west-3": "763104351884",
@@ -69,6 +75,8 @@
6975
"ap-south-1": "763104351884",
7076
"ap-southeast-1": "763104351884",
7177
"ap-southeast-2": "763104351884",
78+
"cn-north-1": "727897471807",
79+
"cn-northwest-1": "727897471807",
7280
"eu-central-1": "763104351884",
7381
"eu-west-1": "763104351884",
7482
"eu-west-3": "763104351884",
@@ -89,6 +97,8 @@
8997
"ap-south-1": "763104351884",
9098
"ap-southeast-1": "763104351884",
9199
"ap-southeast-2": "763104351884",
100+
"cn-north-1": "727897471807",
101+
"cn-northwest-1": "727897471807",
92102
"eu-central-1": "763104351884",
93103
"eu-west-1": "763104351884",
94104
"eu-west-3": "763104351884",
@@ -109,6 +119,8 @@
109119
"ap-south-1": "763104351884",
110120
"ap-southeast-1": "763104351884",
111121
"ap-southeast-2": "763104351884",
122+
"cn-north-1": "727897471807",
123+
"cn-northwest-1": "727897471807",
112124
"eu-central-1": "763104351884",
113125
"eu-west-1": "763104351884",
114126
"eu-west-3": "763104351884",
@@ -129,6 +141,8 @@
129141
"ap-south-1": "763104351884",
130142
"ap-southeast-1": "763104351884",
131143
"ap-southeast-2": "763104351884",
144+
"cn-north-1": "727897471807",
145+
"cn-northwest-1": "727897471807",
132146
"eu-central-1": "763104351884",
133147
"eu-west-1": "763104351884",
134148
"eu-west-3": "763104351884",
@@ -149,6 +163,8 @@
149163
"ap-south-1": "763104351884",
150164
"ap-southeast-1": "763104351884",
151165
"ap-southeast-2": "763104351884",
166+
"cn-north-1": "727897471807",
167+
"cn-northwest-1": "727897471807",
152168
"eu-central-1": "763104351884",
153169
"eu-west-1": "763104351884",
154170
"eu-west-3": "763104351884",

‎src/sagemaker/image_uri_config/huggingface-llm-neuronx.json‎

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@
1818
"ap-southeast-1": "763104351884",
1919
"ap-southeast-2": "763104351884",
2020
"ap-southeast-4": "457447274322",
21+
"cn-north-1": "727897471807",
22+
"cn-northwest-1": "727897471807",
2123
"eu-central-1": "763104351884",
2224
"eu-central-2": "380420809688",
2325
"eu-south-2": "503227376785",
@@ -27,6 +29,8 @@
2729
"sa-east-1": "763104351884",
2830
"us-east-1": "763104351884",
2931
"us-east-2": "763104351884",
32+
"us-gov-east-1": "446045086412",
33+
"us-gov-west-1": "442386744353",
3034
"us-west-2": "763104351884",
3135
"ca-west-1": "204538143572"
3236
},
@@ -47,6 +51,8 @@
4751
"ap-southeast-1": "763104351884",
4852
"ap-southeast-2": "763104351884",
4953
"ap-southeast-4": "457447274322",
54+
"cn-north-1": "727897471807",
55+
"cn-northwest-1": "727897471807",
5056
"eu-central-1": "763104351884",
5157
"eu-central-2": "380420809688",
5258
"eu-south-2": "503227376785",
@@ -56,6 +62,8 @@
5662
"sa-east-1": "763104351884",
5763
"us-east-1": "763104351884",
5864
"us-east-2": "763104351884",
65+
"us-gov-east-1": "446045086412",
66+
"us-gov-west-1": "442386744353",
5967
"us-west-2": "763104351884",
6068
"ca-west-1": "204538143572"
6169
},
@@ -76,6 +84,8 @@
7684
"ap-southeast-1": "763104351884",
7785
"ap-southeast-2": "763104351884",
7886
"ap-southeast-4": "457447274322",
87+
"cn-north-1": "727897471807",
88+
"cn-northwest-1": "727897471807",
7989
"eu-central-1": "763104351884",
8090
"eu-central-2": "380420809688",
8191
"eu-south-2": "503227376785",
@@ -85,6 +95,8 @@
8595
"sa-east-1": "763104351884",
8696
"us-east-1": "763104351884",
8797
"us-east-2": "763104351884",
98+
"us-gov-east-1": "446045086412",
99+
"us-gov-west-1": "442386744353",
88100
"us-west-2": "763104351884",
89101
"ca-west-1": "204538143572"
90102
},
@@ -105,6 +117,8 @@
105117
"ap-southeast-1": "763104351884",
106118
"ap-southeast-2": "763104351884",
107119
"ap-southeast-4": "457447274322",
120+
"cn-north-1": "727897471807",
121+
"cn-northwest-1": "727897471807",
108122
"eu-central-1": "763104351884",
109123
"eu-central-2": "380420809688",
110124
"eu-south-2": "503227376785",
@@ -114,6 +128,8 @@
114128
"sa-east-1": "763104351884",
115129
"us-east-1": "763104351884",
116130
"us-east-2": "763104351884",
131+
"us-gov-east-1": "446045086412",
132+
"us-gov-west-1": "442386744353",
117133
"us-west-2": "763104351884",
118134
"ca-west-1": "204538143572"
119135
},
@@ -134,6 +150,8 @@
134150
"ap-southeast-1": "763104351884",
135151
"ap-southeast-2": "763104351884",
136152
"ap-southeast-4": "457447274322",
153+
"cn-north-1": "727897471807",
154+
"cn-northwest-1": "727897471807",
137155
"eu-central-1": "763104351884",
138156
"eu-central-2": "380420809688",
139157
"eu-south-2": "503227376785",
@@ -143,6 +161,8 @@
143161
"sa-east-1": "763104351884",
144162
"us-east-1": "763104351884",
145163
"us-east-2": "763104351884",
164+
"us-gov-east-1": "446045086412",
165+
"us-gov-west-1": "442386744353",
146166
"us-west-2": "763104351884",
147167
"ca-west-1": "204538143572"
148168
},
@@ -163,6 +183,8 @@
163183
"ap-southeast-1": "763104351884",
164184
"ap-southeast-2": "763104351884",
165185
"ap-southeast-4": "457447274322",
186+
"cn-north-1": "727897471807",
187+
"cn-northwest-1": "727897471807",
166188
"eu-central-1": "763104351884",
167189
"eu-central-2": "380420809688",
168190
"eu-south-2": "503227376785",
@@ -172,6 +194,8 @@
172194
"sa-east-1": "763104351884",
173195
"us-east-1": "763104351884",
174196
"us-east-2": "763104351884",
197+
"us-gov-east-1": "446045086412",
198+
"us-gov-west-1": "442386744353",
175199
"us-west-2": "763104351884",
176200
"ca-west-1": "204538143572"
177201
},
@@ -192,6 +216,8 @@
192216
"ap-southeast-1": "763104351884",
193217
"ap-southeast-2": "763104351884",
194218
"ap-southeast-4": "457447274322",
219+
"cn-north-1": "727897471807",
220+
"cn-northwest-1": "727897471807",
195221
"eu-central-1": "763104351884",
196222
"eu-central-2": "380420809688",
197223
"eu-south-2": "503227376785",
@@ -221,6 +247,8 @@
221247
"ap-southeast-1": "763104351884",
222248
"ap-southeast-2": "763104351884",
223249
"ap-southeast-4": "457447274322",
250+
"cn-north-1": "727897471807",
251+
"cn-northwest-1": "727897471807",
224252
"eu-central-1": "763104351884",
225253
"eu-central-2": "380420809688",
226254
"eu-south-2": "503227376785",
@@ -230,6 +258,8 @@
230258
"sa-east-1": "763104351884",
231259
"us-east-1": "763104351884",
232260
"us-east-2": "763104351884",
261+
"us-gov-east-1": "446045086412",
262+
"us-gov-west-1": "442386744353",
233263
"us-west-2": "763104351884",
234264
"ca-west-1": "204538143572"
235265
},
@@ -250,6 +280,8 @@
250280
"ap-southeast-1": "763104351884",
251281
"ap-southeast-2": "763104351884",
252282
"ap-southeast-4": "457447274322",
283+
"cn-north-1": "727897471807",
284+
"cn-northwest-1": "727897471807",
253285
"eu-central-1": "763104351884",
254286
"eu-central-2": "380420809688",
255287
"eu-south-2": "503227376785",
@@ -259,6 +291,8 @@
259291
"sa-east-1": "763104351884",
260292
"us-east-1": "763104351884",
261293
"us-east-2": "763104351884",
294+
"us-gov-east-1": "446045086412",
295+
"us-gov-west-1": "442386744353",
262296
"us-west-2": "763104351884",
263297
"ca-west-1": "204538143572"
264298
},

‎src/sagemaker/image_uri_config/huggingface-neuron.json‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@
2323
"ap-southeast-1": "763104351884",
2424
"ap-southeast-2": "763104351884",
2525
"ap-southeast-4": "457447274322",
26+
"cn-north-1": "727897471807",
27+
"cn-northwest-1": "727897471807",
2628
"eu-central-1": "763104351884",
2729
"eu-central-2": "380420809688",
2830
"eu-south-2": "503227376785",
@@ -32,6 +34,8 @@
3234
"sa-east-1": "763104351884",
3335
"us-east-1": "763104351884",
3436
"us-east-2": "763104351884",
37+
"us-gov-east-1": "446045086412",
38+
"us-gov-west-1": "442386744353",
3539
"us-west-2": "763104351884",
3640
"ca-west-1": "204538143572"
3741
},

‎src/sagemaker/image_uri_config/huggingface-neuronx.json‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,8 @@
2525
"ap-southeast-1": "763104351884",
2626
"ap-southeast-2": "763104351884",
2727
"ap-southeast-4": "457447274322",
28+
"cn-north-1": "727897471807",
29+
"cn-northwest-1": "727897471807",
2830
"eu-central-1": "763104351884",
2931
"eu-central-2": "380420809688",
3032
"eu-south-2": "503227376785",
@@ -34,6 +36,8 @@
3436
"sa-east-1": "763104351884",
3537
"us-east-1": "763104351884",
3638
"us-east-2": "763104351884",
39+
"us-gov-east-1": "446045086412",
40+
"us-gov-west-1": "442386744353",
3741
"us-west-2": "763104351884",
3842
"ca-west-1": "204538143572"
3943
},
@@ -61,6 +65,8 @@
6165
"ap-southeast-1": "763104351884",
6266
"ap-southeast-2": "763104351884",
6367
"ap-southeast-4": "457447274322",
68+
"cn-north-1": "727897471807",
69+
"cn-northwest-1": "727897471807",
6470
"eu-central-1": "763104351884",
6571
"eu-central-2": "380420809688",
6672
"eu-south-2": "503227376785",
@@ -70,6 +76,8 @@
7076
"sa-east-1": "763104351884",
7177
"us-east-1": "763104351884",
7278
"us-east-2": "763104351884",
79+
"us-gov-east-1": "446045086412",
80+
"us-gov-west-1": "442386744353",
7381
"us-west-2": "763104351884",
7482
"ca-west-1": "204538143572"
7583
},
@@ -97,6 +105,8 @@
97105
"ap-southeast-1": "763104351884",
98106
"ap-southeast-2": "763104351884",
99107
"ap-southeast-4": "457447274322",
108+
"cn-north-1": "727897471807",
109+
"cn-northwest-1": "727897471807",
100110
"eu-central-1": "763104351884",
101111
"eu-central-2": "380420809688",
102112
"eu-south-2": "503227376785",
@@ -106,6 +116,8 @@
106116
"sa-east-1": "763104351884",
107117
"us-east-1": "763104351884",
108118
"us-east-2": "763104351884",
119+
"us-gov-east-1": "446045086412",
120+
"us-gov-west-1": "442386744353",
109121
"us-west-2": "763104351884",
110122
"ca-west-1": "204538143572"
111123
},

‎src/sagemaker/image_uri_config/huggingface-training-compiler.json‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,8 @@
7070
"ap-southeast-3": "907027046896",
7171
"ap-southeast-4": "457447274322",
7272
"ca-central-1": "763104351884",
73+
"cn-north-1": "727897471807",
74+
"cn-northwest-1": "727897471807",
7375
"eu-central-1": "763104351884",
7476
"eu-central-2": "380420809688",
7577
"eu-north-1": "763104351884",
@@ -83,6 +85,8 @@
8385
"sa-east-1": "763104351884",
8486
"us-east-1": "763104351884",
8587
"us-east-2": "763104351884",
88+
"us-gov-east-1": "446045086412",
89+
"us-gov-west-1": "442386744353",
8690
"us-west-1": "763104351884",
8791
"us-west-2": "763104351884",
8892
"ca-west-1": "204538143572"
@@ -110,6 +114,8 @@
110114
"ap-southeast-3": "907027046896",
111115
"ap-southeast-4": "457447274322",
112116
"ca-central-1": "763104351884",
117+
"cn-north-1": "727897471807",
118+
"cn-northwest-1": "727897471807",
113119
"eu-central-1": "763104351884",
114120
"eu-central-2": "380420809688",
115121
"eu-north-1": "763104351884",
@@ -123,6 +129,8 @@
123129
"sa-east-1": "763104351884",
124130
"us-east-1": "763104351884",
125131
"us-east-2": "763104351884",
132+
"us-gov-east-1": "446045086412",
133+
"us-gov-west-1": "442386744353",
126134
"us-west-1": "763104351884",
127135
"us-west-2": "763104351884",
128136
"ca-west-1": "204538143572"
@@ -168,6 +176,8 @@
168176
"sa-east-1": "763104351884",
169177
"us-east-1": "763104351884",
170178
"us-east-2": "763104351884",
179+
"us-gov-east-1": "446045086412",
180+
"us-gov-west-1": "442386744353",
171181
"us-west-1": "763104351884",
172182
"us-west-2": "763104351884",
173183
"ca-west-1": "204538143572"

‎src/sagemaker/jumpstart/constants.py‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,12 @@
176176
JumpStartLaunchedRegionInfo(
177177
region_name="cn-north-1",
178178
content_bucket="jumpstart-cache-prod-cn-north-1",
179+
gated_content_bucket="jumpstart-private-cache-prod-cn-north-1",
180+
),
181+
JumpStartLaunchedRegionInfo(
182+
region_name="cn-northwest-1",
183+
content_bucket="jumpstart-cache-prod-cn-northwest-1",
184+
gated_content_bucket="jumpstart-private-cache-prod-cn-northwest-1",
179185
),
180186
JumpStartLaunchedRegionInfo(
181187
region_name="il-central-1",

0 commit comments

Comments
 (0)