Skip to content

Commit 0ae2457

Browse files
huggingface-tei dlc image_uri (#5174)
Co-authored-by: pintaoz-aws <[email protected]>
1 parent 6809486 commit 0ae2457

File tree

2 files changed

+98
-2
lines changed

2 files changed

+98
-2
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
}

0 commit comments

Comments
 (0)