Skip to content

Commit c4f5cd7

Browse files
committed
chore: fix tei tests
1 parent c93a712 commit c4f5cd7

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
@@ -25,12 +25,14 @@
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",
2727
"1.7.0": "2.0.1-tei1.7.0-gpu-py310-cu122-ubuntu22.04",
28+
"1.8.0": "2.0.1-tei1.8.0-gpu-py310-cu122-ubuntu22.04",
2829
},
2930
"cpu": {
3031
"1.2.3": "2.0.1-tei1.2.3-cpu-py310-ubuntu22.04",
3132
"1.4.0": "2.0.1-tei1.4.0-cpu-py310-ubuntu22.04",
3233
"1.6.0": "2.0.1-tei1.6.0-cpu-py310-ubuntu22.04",
3334
"1.7.0": "2.0.1-tei1.7.0-cpu-py310-ubuntu22.04",
35+
"1.8.0": "2.0.1-tei1.8.0-cpu-py310-ubuntu22.04",
3436
},
3537
}
3638
HF_VERSIONS_MAPPING = {

0 commit comments

Comments
 (0)