Skip to content

Commit e2ea4ff

Browse files
pagezyhfpintaoz-aws
pagezyhf
and
pintaoz-aws
authored
huggingface-neuronx dlc image_uri (#5172)
* huggingface-neuronx dlc image_uri * huggingface-neuronx inference dlc --------- Co-authored-by: pintaoz-aws <[email protected]>
1 parent 0ae2457 commit e2ea4ff

File tree

1 file changed

+134
-2
lines changed

1 file changed

+134
-2
lines changed

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

Lines changed: 134 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,9 @@
66
"version_aliases": {
77
"4.28": "4.28.1",
88
"4.34": "4.34.1",
9-
"4.36": "4.36.2"
9+
"4.36": "4.36.2",
10+
"4.43": "4.43.2",
11+
"4.48": "4.48.1"
1012
},
1113
"versions": {
1214
"4.28.1": {
@@ -137,6 +139,92 @@
137139
"sdk2.18.0"
138140
]
139141
}
142+
},
143+
"4.43.2": {
144+
"version_aliases": {
145+
"pytorch2.1": "pytorch2.1.2"
146+
},
147+
"pytorch2.1.2": {
148+
"py_versions": [
149+
"py310"
150+
],
151+
"repository": "huggingface-pytorch-inference-neuronx",
152+
"registries": {
153+
"ap-northeast-1": "763104351884",
154+
"ap-south-1": "763104351884",
155+
"ap-south-2": "772153158452",
156+
"ap-southeast-1": "763104351884",
157+
"ap-southeast-2": "763104351884",
158+
"ap-southeast-4": "457447274322",
159+
"ap-southeast-5": "550225433462",
160+
"ap-southeast-7": "590183813437",
161+
"cn-north-1": "727897471807",
162+
"cn-northwest-1": "727897471807",
163+
"eu-central-1": "763104351884",
164+
"eu-central-2": "380420809688",
165+
"eu-south-2": "503227376785",
166+
"eu-west-1": "763104351884",
167+
"eu-west-3": "763104351884",
168+
"il-central-1": "780543022126",
169+
"mx-central-1":"637423239942",
170+
"sa-east-1": "763104351884",
171+
"us-east-1": "763104351884",
172+
"us-east-2": "763104351884",
173+
"us-gov-east-1": "446045086412",
174+
"us-gov-west-1": "442386744353",
175+
"us-west-2": "763104351884",
176+
"ca-west-1": "204538143572"
177+
},
178+
"container_version": {
179+
"inf": "ubuntu20.04"
180+
},
181+
"sdk_versions": [
182+
"sdk2.20.0"
183+
]
184+
}
185+
},
186+
"4.48.1": {
187+
"version_aliases": {
188+
"pytorch2.1": "pytorch2.1.2"
189+
},
190+
"pytorch2.1.2": {
191+
"py_versions": [
192+
"py310"
193+
],
194+
"repository": "huggingface-pytorch-inference-neuronx",
195+
"registries": {
196+
"ap-northeast-1": "763104351884",
197+
"ap-south-1": "763104351884",
198+
"ap-south-2": "772153158452",
199+
"ap-southeast-1": "763104351884",
200+
"ap-southeast-2": "763104351884",
201+
"ap-southeast-4": "457447274322",
202+
"ap-southeast-5": "550225433462",
203+
"ap-southeast-7": "590183813437",
204+
"cn-north-1": "727897471807",
205+
"cn-northwest-1": "727897471807",
206+
"eu-central-1": "763104351884",
207+
"eu-central-2": "380420809688",
208+
"eu-south-2": "503227376785",
209+
"eu-west-1": "763104351884",
210+
"eu-west-3": "763104351884",
211+
"il-central-1": "780543022126",
212+
"mx-central-1":"637423239942",
213+
"sa-east-1": "763104351884",
214+
"us-east-1": "763104351884",
215+
"us-east-2": "763104351884",
216+
"us-gov-east-1": "446045086412",
217+
"us-gov-west-1": "442386744353",
218+
"us-west-2": "763104351884",
219+
"ca-west-1": "204538143572"
220+
},
221+
"container_version": {
222+
"inf": "ubuntu20.04"
223+
},
224+
"sdk_versions": [
225+
"sdk2.20.0"
226+
]
227+
}
140228
}
141229
}
142230
},
@@ -147,7 +235,8 @@
147235
"version_aliases": {
148236
"4.28": "4.28.1",
149237
"4.34": "4.34.1",
150-
"4.36": "4.36.2"
238+
"4.36": "4.36.2",
239+
"4.43": "4.43.2"
151240
},
152241
"versions": {
153242
"4.28.1": {
@@ -365,6 +454,49 @@
365454
"sdk2.18.0"
366455
]
367456
}
457+
},
458+
"4.43.2": {
459+
"version_aliases": {
460+
"pytorch2.1": "pytorch2.1.2"
461+
},
462+
"pytorch2.1.2": {
463+
"py_versions": [
464+
"py310"
465+
],
466+
"repository": "huggingface-pytorch-inference-neuronx",
467+
"registries": {
468+
"ap-northeast-1": "763104351884",
469+
"ap-south-1": "763104351884",
470+
"ap-south-2": "772153158452",
471+
"ap-southeast-1": "763104351884",
472+
"ap-southeast-2": "763104351884",
473+
"ap-southeast-4": "457447274322",
474+
"ap-southeast-5": "550225433462",
475+
"ap-southeast-7": "590183813437",
476+
"cn-north-1": "727897471807",
477+
"cn-northwest-1": "727897471807",
478+
"eu-central-1": "763104351884",
479+
"eu-central-2": "380420809688",
480+
"eu-south-2": "503227376785",
481+
"eu-west-1": "763104351884",
482+
"eu-west-3": "763104351884",
483+
"il-central-1": "780543022126",
484+
"mx-central-1":"637423239942",
485+
"sa-east-1": "763104351884",
486+
"us-east-1": "763104351884",
487+
"us-east-2": "763104351884",
488+
"us-gov-east-1": "446045086412",
489+
"us-gov-west-1": "442386744353",
490+
"us-west-2": "763104351884",
491+
"ca-west-1": "204538143572"
492+
},
493+
"container_version": {
494+
"inf": "ubuntu20.04"
495+
},
496+
"sdk_versions": [
497+
"sdk2.20.0"
498+
]
499+
}
368500
}
369501
}
370502
}

0 commit comments

Comments
 (0)