Skip to content

Commit 4cc17e0

Browse files
authored
add TEI 1.8.2 (#5305)
* add TEI 1.8.2 * add test
1 parent 4cc4944 commit 4cc17e0

File tree

3 files changed

+98
-2
lines changed

3 files changed

+98
-2
lines changed

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

Lines changed: 48 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,54 @@
244244
"container_version": {
245245
"cpu": "ubuntu22.04"
246246
}
247+
},
248+
"1.8.2":{
249+
"py_versions": [
250+
"py310"
251+
],
252+
"registries": {
253+
"af-south-1": "510948584623",
254+
"ap-east-1": "651117190479",
255+
"ap-northeast-1": "354813040037",
256+
"ap-northeast-2": "366743142698",
257+
"ap-northeast-3": "867004704886",
258+
"ap-south-1": "720646828776",
259+
"ap-south-2": "628508329040",
260+
"ap-southeast-1": "121021644041",
261+
"ap-southeast-2": "783357654285",
262+
"ap-southeast-3": "951798379941",
263+
"ap-southeast-4": "106583098589",
264+
"ca-central-1": "341280168497",
265+
"ca-west-1": "190319476487",
266+
"cn-north-1": "450853457545",
267+
"cn-northwest-1": "451049120500",
268+
"eu-central-1": "492215442770",
269+
"eu-central-2": "680994064768",
270+
"eu-north-1": "662702820516",
271+
"eu-south-1": "978288397137",
272+
"eu-south-2": "104374241257",
273+
"eu-west-1": "141502667606",
274+
"eu-west-2": "764974769150",
275+
"eu-west-3": "659782779980",
276+
"il-central-1": "898809789911",
277+
"me-central-1": "272398656194",
278+
"me-south-1": "801668240914",
279+
"sa-east-1": "737474898029",
280+
"us-east-1": "683313688378",
281+
"us-east-2": "257758044811",
282+
"us-gov-east-1": "237065988967",
283+
"us-gov-west-1": "414596584902",
284+
"us-iso-east-1": "833128469047",
285+
"us-isob-east-1": "281123927165",
286+
"us-west-1": "746614075791",
287+
"us-west-2": "246618743249"
288+
},
289+
"tag_prefix": "2.0.1-tei1.8.2",
290+
"repository": "tei-cpu",
291+
"container_version": {
292+
"cpu": "ubuntu22.04"
293+
}
247294
}
248295
}
249296
}
250-
}
297+
}

src/sagemaker/image_uri_config/huggingface-tei.json

Lines changed: 48 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,54 @@
244244
"container_version": {
245245
"gpu": "cu122-ubuntu22.04"
246246
}
247+
},
248+
"1.8.2": {
249+
"py_versions": [
250+
"py310"
251+
],
252+
"registries": {
253+
"af-south-1": "510948584623",
254+
"ap-east-1": "651117190479",
255+
"ap-northeast-1": "354813040037",
256+
"ap-northeast-2": "366743142698",
257+
"ap-northeast-3": "867004704886",
258+
"ap-south-1": "720646828776",
259+
"ap-south-2": "628508329040",
260+
"ap-southeast-1": "121021644041",
261+
"ap-southeast-2": "783357654285",
262+
"ap-southeast-3": "951798379941",
263+
"ap-southeast-4": "106583098589",
264+
"ca-central-1": "341280168497",
265+
"ca-west-1": "190319476487",
266+
"cn-north-1": "450853457545",
267+
"cn-northwest-1": "451049120500",
268+
"eu-central-1": "492215442770",
269+
"eu-central-2": "680994064768",
270+
"eu-north-1": "662702820516",
271+
"eu-south-1": "978288397137",
272+
"eu-south-2": "104374241257",
273+
"eu-west-1": "141502667606",
274+
"eu-west-2": "764974769150",
275+
"eu-west-3": "659782779980",
276+
"il-central-1": "898809789911",
277+
"me-central-1": "272398656194",
278+
"me-south-1": "801668240914",
279+
"sa-east-1": "737474898029",
280+
"us-east-1": "683313688378",
281+
"us-east-2": "257758044811",
282+
"us-gov-east-1": "237065988967",
283+
"us-gov-west-1": "414596584902",
284+
"us-iso-east-1": "833128469047",
285+
"us-isob-east-1": "281123927165",
286+
"us-west-1": "746614075791",
287+
"us-west-2": "246618743249"
288+
},
289+
"tag_prefix": "2.0.1-tei1.8.2",
290+
"repository": "tei",
291+
"container_version": {
292+
"gpu": "cu122-ubuntu22.04"
293+
}
247294
}
248295
}
249296
}
250-
}
297+
}

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,13 +26,15 @@
2626
"1.6.0": "2.0.1-tei1.6.0-gpu-py310-cu122-ubuntu22.04",
2727
"1.7.0": "2.0.1-tei1.7.0-gpu-py310-cu122-ubuntu22.04",
2828
"1.8.0": "2.0.1-tei1.8.0-gpu-py310-cu122-ubuntu22.04",
29+
"1.8.2": "2.0.1-tei1.8.2-gpu-py310-cu122-ubuntu22.04",
2930
},
3031
"cpu": {
3132
"1.2.3": "2.0.1-tei1.2.3-cpu-py310-ubuntu22.04",
3233
"1.4.0": "2.0.1-tei1.4.0-cpu-py310-ubuntu22.04",
3334
"1.6.0": "2.0.1-tei1.6.0-cpu-py310-ubuntu22.04",
3435
"1.7.0": "2.0.1-tei1.7.0-cpu-py310-ubuntu22.04",
3536
"1.8.0": "2.0.1-tei1.8.0-cpu-py310-ubuntu22.04",
37+
"1.8.2": "2.0.1-tei1.8.2-cpu-py310-ubuntu22.04",
3638
},
3739
}
3840
HF_VERSIONS_MAPPING = {

0 commit comments

Comments
 (0)