Skip to content

Commit 59539b6

Browse files
author
pintaoz
committed
Fix test_huggingface_tei_uris()
1 parent 84852dd commit 59539b6

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
@@ -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)