Skip to content

Commit f3e221b

Browse files
committed
chore: add tei 1.6.0 to tei mapping in tests
1 parent 4d0ff26 commit f3e221b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,12 @@
2323
"gpu": {
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",
26+
"1.6.0": "2.0.1-tei1.6.0-gpu-py310-cu122-ubuntu22.04",
2627
},
2728
"cpu": {
2829
"1.2.3": "2.0.1-tei1.2.3-cpu-py310-ubuntu22.04",
2930
"1.4.0": "2.0.1-tei1.4.0-cpu-py310-ubuntu22.04",
31+
"1.6.0": "2.0.1-tei1.6.0-cpu-py310-ubuntu22.04",
3032
},
3133
}
3234
HF_VERSIONS_MAPPING = {

0 commit comments

Comments
 (0)