Skip to content

Commit 27d493f

Browse files
varunmorisvarunmoris
committed
chore: add new huggingface images
1 parent b8771e3 commit 27d493f

File tree

3 files changed

+201
-4
lines changed

3 files changed

+201
-4
lines changed

‎src/sagemaker/image_uri_config/huggingface-tei-cpu.json

Lines changed: 49 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,8 @@
66
"version_aliases": {
77
"1.2": "1.2.3",
88
"1.4": "1.4.0",
9-
"1.6": "1.6.0"
9+
"1.6": "1.6.0",
10+
"1.7": "1.7.0"
1011
},
1112
"versions": {
1213
"1.2.3": {
@@ -149,6 +150,53 @@
149150
"container_version": {
150151
"cpu": "ubuntu22.04"
151152
}
153+
},
154+
"1.7.0":{
155+
"py_versions": [
156+
"py310"
157+
],
158+
"registries": {
159+
"af-south-1": "510948584623",
160+
"ap-east-1": "651117190479",
161+
"ap-northeast-1": "354813040037",
162+
"ap-northeast-2": "366743142698",
163+
"ap-northeast-3": "867004704886",
164+
"ap-south-1": "720646828776",
165+
"ap-south-2": "628508329040",
166+
"ap-southeast-1": "121021644041",
167+
"ap-southeast-2": "783357654285",
168+
"ap-southeast-3": "951798379941",
169+
"ap-southeast-4": "106583098589",
170+
"ca-central-1": "341280168497",
171+
"ca-west-1": "190319476487",
172+
"cn-north-1": "450853457545",
173+
"cn-northwest-1": "451049120500",
174+
"eu-central-1": "492215442770",
175+
"eu-central-2": "680994064768",
176+
"eu-north-1": "662702820516",
177+
"eu-south-1": "978288397137",
178+
"eu-south-2": "104374241257",
179+
"eu-west-1": "141502667606",
180+
"eu-west-2": "764974769150",
181+
"eu-west-3": "659782779980",
182+
"il-central-1": "898809789911",
183+
"me-central-1": "272398656194",
184+
"me-south-1": "801668240914",
185+
"sa-east-1": "737474898029",
186+
"us-east-1": "683313688378",
187+
"us-east-2": "257758044811",
188+
"us-gov-east-1": "237065988967",
189+
"us-gov-west-1": "414596584902",
190+
"us-iso-east-1": "833128469047",
191+
"us-isob-east-1": "281123927165",
192+
"us-west-1": "746614075791",
193+
"us-west-2": "246618743249"
194+
},
195+
"tag_prefix": "2.0.1-tei1.7.0",
196+
"repository": "tei-cpu",
197+
"container_version": {
198+
"cpu": "ubuntu22.04"
199+
}
152200
}
153201
}
154202
}

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

Lines changed: 49 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,8 @@
66
"version_aliases": {
77
"1.2": "1.2.3",
88
"1.4": "1.4.0",
9-
"1.6": "1.6.0"
9+
"1.6": "1.6.0",
10+
"1.7": "1.7.0"
1011
},
1112
"versions": {
1213
"1.2.3": {
@@ -149,6 +150,53 @@
149150
"container_version": {
150151
"gpu": "cu122-ubuntu22.04"
151152
}
153+
},
154+
"1.7.0": {
155+
"py_versions": [
156+
"py310"
157+
],
158+
"registries": {
159+
"af-south-1": "510948584623",
160+
"ap-east-1": "651117190479",
161+
"ap-northeast-1": "354813040037",
162+
"ap-northeast-2": "366743142698",
163+
"ap-northeast-3": "867004704886",
164+
"ap-south-1": "720646828776",
165+
"ap-south-2": "628508329040",
166+
"ap-southeast-1": "121021644041",
167+
"ap-southeast-2": "783357654285",
168+
"ap-southeast-3": "951798379941",
169+
"ap-southeast-4": "106583098589",
170+
"ca-central-1": "341280168497",
171+
"ca-west-1": "190319476487",
172+
"cn-north-1": "450853457545",
173+
"cn-northwest-1": "451049120500",
174+
"eu-central-1": "492215442770",
175+
"eu-central-2": "680994064768",
176+
"eu-north-1": "662702820516",
177+
"eu-south-1": "978288397137",
178+
"eu-south-2": "104374241257",
179+
"eu-west-1": "141502667606",
180+
"eu-west-2": "764974769150",
181+
"eu-west-3": "659782779980",
182+
"il-central-1": "898809789911",
183+
"me-central-1": "272398656194",
184+
"me-south-1": "801668240914",
185+
"sa-east-1": "737474898029",
186+
"us-east-1": "683313688378",
187+
"us-east-2": "257758044811",
188+
"us-gov-east-1": "237065988967",
189+
"us-gov-west-1": "414596584902",
190+
"us-iso-east-1": "833128469047",
191+
"us-isob-east-1": "281123927165",
192+
"us-west-1": "746614075791",
193+
"us-west-2": "246618743249"
194+
},
195+
"tag_prefix": "2.0.1-tei1.7.0",
196+
"repository": "tei",
197+
"container_version": {
198+
"gpu": "cu122-ubuntu22.04"
199+
}
152200
}
153201
}
154202
}

‎src/sagemaker/image_uri_config/huggingface.json

Lines changed: 103 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,8 @@
1616
"4.36": "4.36.0",
1717
"4.46": "4.46.1",
1818
"4.48": "4.48.0",
19-
"4.49": "4.49.0"
19+
"4.49": "4.49.0",
20+
"4.53": "4.53.1"
2021
},
2122
"versions": {
2223
"4.4.2": {
@@ -1162,6 +1163,53 @@
11621163
"gpu": "cu124-ubuntu22.04"
11631164
}
11641165
}
1166+
},
1167+
"4.53.1": {
1168+
"version_aliases": {
1169+
"pytorch2.6": "pytorch2.6.0"
1170+
},
1171+
"pytorch2.5.1": {
1172+
"py_versions": [
1173+
"py312"
1174+
],
1175+
"registries": {
1176+
"af-south-1": "626614931356",
1177+
"il-central-1": "780543022126",
1178+
"ap-east-1": "871362719292",
1179+
"ap-northeast-1": "763104351884",
1180+
"ap-northeast-2": "763104351884",
1181+
"ap-northeast-3": "364406365360",
1182+
"ap-south-1": "763104351884",
1183+
"ap-southeast-1": "763104351884",
1184+
"ap-southeast-2": "763104351884",
1185+
"ap-southeast-3": "907027046896",
1186+
"ca-central-1": "763104351884",
1187+
"cn-north-1": "727897471807",
1188+
"cn-northwest-1": "727897471807",
1189+
"eu-central-1": "763104351884",
1190+
"eu-north-1": "763104351884",
1191+
"eu-west-1": "763104351884",
1192+
"eu-west-2": "763104351884",
1193+
"eu-west-3": "763104351884",
1194+
"eu-south-1": "692866216735",
1195+
"me-south-1": "217643126080",
1196+
"me-central-1": "914824155844",
1197+
"sa-east-1": "763104351884",
1198+
"us-east-1": "763104351884",
1199+
"us-east-2": "763104351884",
1200+
"us-gov-east-1": "446045086412",
1201+
"us-gov-west-1": "442386744353",
1202+
"us-iso-east-1": "886529160074",
1203+
"us-isob-east-1": "094389454867",
1204+
"us-west-1": "763104351884",
1205+
"us-west-2": "763104351884",
1206+
"ca-west-1": "204538143572"
1207+
},
1208+
"repository": "huggingface-pytorch-training",
1209+
"container_version": {
1210+
"gpu": "cu126-ubuntu22.04"
1211+
}
1212+
}
11651213
}
11661214
}
11671215
},
@@ -1179,7 +1227,8 @@
11791227
"4.26": "4.26.0",
11801228
"4.28": "4.28.1",
11811229
"4.37": "4.37.0",
1182-
"4.49": "4.49.0"
1230+
"4.49": "4.49.0",
1231+
"4.53": "4.53.1"
11831232
},
11841233
"versions": {
11851234
"4.6.1": {
@@ -2132,6 +2181,58 @@
21322181
"cpu": "ubuntu22.04"
21332182
}
21342183
}
2184+
},
2185+
"4.53.1": {
2186+
"version_aliases": {
2187+
"pytorch2.6": "pytorch2.6.0"
2188+
},
2189+
"pytorch2.6.0": {
2190+
"py_versions": [
2191+
"py312"
2192+
],
2193+
"registries": {
2194+
"af-south-1": "626614931356",
2195+
"il-central-1": "780543022126",
2196+
"ap-east-1": "871362719292",
2197+
"ap-northeast-1": "763104351884",
2198+
"ap-northeast-2": "763104351884",
2199+
"ap-northeast-3": "364406365360",
2200+
"ap-south-1": "763104351884",
2201+
"ap-south-2": "772153158452",
2202+
"ap-southeast-1": "763104351884",
2203+
"ap-southeast-2": "763104351884",
2204+
"ap-southeast-3": "907027046896",
2205+
"ap-southeast-4": "457447274322",
2206+
"ca-central-1": "763104351884",
2207+
"cn-north-1": "727897471807",
2208+
"cn-northwest-1": "727897471807",
2209+
"eu-central-1": "763104351884",
2210+
"eu-central-2": "380420809688",
2211+
"eu-north-1": "763104351884",
2212+
"eu-west-1": "763104351884",
2213+
"eu-west-2": "763104351884",
2214+
"eu-west-3": "763104351884",
2215+
"eu-south-1": "692866216735",
2216+
"eu-south-2": "503227376785",
2217+
"me-south-1": "217643126080",
2218+
"me-central-1": "914824155844",
2219+
"sa-east-1": "763104351884",
2220+
"us-east-1": "763104351884",
2221+
"us-east-2": "763104351884",
2222+
"us-gov-east-1": "446045086412",
2223+
"us-gov-west-1": "442386744353",
2224+
"us-iso-east-1": "886529160074",
2225+
"us-isob-east-1": "094389454867",
2226+
"us-west-1": "763104351884",
2227+
"us-west-2": "763104351884",
2228+
"ca-west-1": "204538143572"
2229+
},
2230+
"repository": "huggingface-pytorch-inference",
2231+
"container_version": {
2232+
"gpu": "cu124-ubuntu22.04",
2233+
"cpu": "ubuntu22.04"
2234+
}
2235+
}
21352236
}
21362237
}
21372238
}

0 commit comments

Comments
 (0)