Skip to content

Commit a47d67c

Browse files
authored
Merge branch 'master' into master-update-default-configs
2 parents 716e71b + 7825dc9 commit a47d67c

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
"inf2"
55
],
66
"version_aliases": {
7-
"0.0": "0.0.28",
7+
"0.0": "0.0.28"
88
},
99
"versions": {
1010
"0.0.16": {

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,13 @@
2424
"1.2.3": "2.0.1-tei1.2.3-gpu-py310-cu122-ubuntu22.04",
2525
"1.4.0": "2.0.1-tei1.4.0-gpu-py310-cu122-ubuntu22.04",
2626
"1.6.0": "2.0.1-tei1.6.0-gpu-py310-cu122-ubuntu22.04",
27+
"1.7.0": "2.0.1-tei1.7.0-gpu-py310-cu122-ubuntu22.04",
2728
},
2829
"cpu": {
2930
"1.2.3": "2.0.1-tei1.2.3-cpu-py310-ubuntu22.04",
3031
"1.4.0": "2.0.1-tei1.4.0-cpu-py310-ubuntu22.04",
3132
"1.6.0": "2.0.1-tei1.6.0-cpu-py310-ubuntu22.04",
33+
"1.7.0": "2.0.1-tei1.7.0-cpu-py310-ubuntu22.04",
3234
},
3335
}
3436
HF_VERSIONS_MAPPING = {

0 commit comments

Comments
 (0)