Skip to content

Commit a6073d9

Browse files
amzn-choericroot
authored and
root
committed
change: TGI NeuronX 0.0.17 (aws#4390)
1 parent 2fda21e commit a6073d9

File tree

2 files changed

+30
-0
lines changed

2 files changed

+30
-0
lines changed

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

+29
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,35 @@
3535
"container_version": {
3636
"inf2": "ubuntu22.04"
3737
}
38+
},
39+
"0.0.17": {
40+
"py_versions": [
41+
"py310"
42+
],
43+
"registries": {
44+
"ap-northeast-1": "763104351884",
45+
"ap-south-1": "763104351884",
46+
"ap-south-2": "772153158452",
47+
"ap-southeast-1": "763104351884",
48+
"ap-southeast-2": "763104351884",
49+
"ap-southeast-4": "457447274322",
50+
"eu-central-1": "763104351884",
51+
"eu-central-2": "380420809688",
52+
"eu-south-2": "503227376785",
53+
"eu-west-1": "763104351884",
54+
"eu-west-3": "763104351884",
55+
"il-central-1": "780543022126",
56+
"sa-east-1": "763104351884",
57+
"us-east-1": "763104351884",
58+
"us-east-2": "763104351884",
59+
"us-west-2": "763104351884",
60+
"ca-west-1": "204538143572"
61+
},
62+
"tag_prefix": "1.13.1-optimum0.0.17",
63+
"repository": "huggingface-pytorch-tgi-inference",
64+
"container_version": {
65+
"inf2": "ubuntu22.04"
66+
}
3867
}
3968
}
4069
}

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

+1
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
},
3232
"inf2": {
3333
"0.0.16": "1.13.1-optimum0.0.16-neuronx-py310-ubuntu22.04",
34+
"0.0.17": "1.13.1-optimum0.0.17-neuronx-py310-ubuntu22.04",
3435
},
3536
}
3637

0 commit comments

Comments
 (0)