We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4d0ff26 commit f3e221bCopy full SHA for f3e221b
tests/unit/sagemaker/image_uris/test_huggingface_llm.py
@@ -23,10 +23,12 @@
23
"gpu": {
24
"1.2.3": "2.0.1-tei1.2.3-gpu-py310-cu122-ubuntu22.04",
25
"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",
27
},
28
"cpu": {
29
"1.2.3": "2.0.1-tei1.2.3-cpu-py310-ubuntu22.04",
30
"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",
32
33
}
34
HF_VERSIONS_MAPPING = {
0 commit comments