Skip to content

Commit 70dd07f

Browse files
authored
feat: TGI 1.2.0 Image Uri (#4297)
1 parent 3ed0011 commit 70dd07f

File tree

2 files changed

+43
-0
lines changed

2 files changed

+43
-0
lines changed

src/sagemaker/image_uri_config/huggingface-llm.json

Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -218,6 +218,48 @@
218218
"tag_prefix": "2.0.1-tgi1.1.0",
219219
"repository": "huggingface-pytorch-tgi-inference",
220220
"container_version": {"gpu": "cu118-ubuntu20.04"}
221+
},
222+
"1.2.0": {
223+
"py_versions": ["py310"],
224+
"registries": {
225+
"af-south-1": "626614931356",
226+
"il-central-1": "780543022126",
227+
"ap-east-1": "871362719292",
228+
"ap-northeast-1": "763104351884",
229+
"ap-northeast-2": "763104351884",
230+
"ap-northeast-3": "364406365360",
231+
"ap-south-1": "763104351884",
232+
"ap-south-2": "772153158452",
233+
"ap-southeast-1": "763104351884",
234+
"ap-southeast-2": "763104351884",
235+
"ap-southeast-3": "907027046896",
236+
"ap-southeast-4": "457447274322",
237+
"ca-central-1": "763104351884",
238+
"cn-north-1": "727897471807",
239+
"cn-northwest-1": "727897471807",
240+
"eu-central-1": "763104351884",
241+
"eu-central-2": "380420809688",
242+
"eu-north-1": "763104351884",
243+
"eu-west-1": "763104351884",
244+
"eu-west-2": "763104351884",
245+
"eu-west-3": "763104351884",
246+
"eu-south-1": "692866216735",
247+
"eu-south-2": "503227376785",
248+
"me-south-1": "217643126080",
249+
"me-central-1": "914824155844",
250+
"sa-east-1": "763104351884",
251+
"us-east-1": "763104351884",
252+
"us-east-2": "763104351884",
253+
"us-gov-east-1": "446045086412",
254+
"us-gov-west-1": "442386744353",
255+
"us-iso-east-1": "886529160074",
256+
"us-isob-east-1": "094389454867",
257+
"us-west-1": "763104351884",
258+
"us-west-2": "763104351884"
259+
},
260+
"tag_prefix": "2.1.1-tgi1.2.0",
261+
"repository": "huggingface-pytorch-tgi-inference",
262+
"container_version": {"gpu": "cu121-ubuntu20.04"}
221263
}
222264
}
223265
}

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
"0.9.3": "2.0.1-tgi0.9.3-gpu-py39-cu118-ubuntu20.04",
2525
"1.0.3": "2.0.1-tgi1.0.3-gpu-py39-cu118-ubuntu20.04",
2626
"1.1.0": "2.0.1-tgi1.1.0-gpu-py39-cu118-ubuntu20.04",
27+
"1.2.0": "2.1.1-tgi1.2.0-gpu-py310-cu121-ubuntu20.04",
2728
}
2829

2930

0 commit comments

Comments
 (0)