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 84852dd commit 59539b6Copy full SHA for 59539b6
tests/unit/sagemaker/image_uris/test_huggingface_llm.py
@@ -24,11 +24,13 @@
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
+ "1.7.0": "2.0.1-tei1.7.0-gpu-py310-cu122-ubuntu22.04",
28
},
29
"cpu": {
30
"1.2.3": "2.0.1-tei1.2.3-cpu-py310-ubuntu22.04",
31
"1.4.0": "2.0.1-tei1.4.0-cpu-py310-ubuntu22.04",
32
"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",
34
35
}
36
HF_VERSIONS_MAPPING = {
0 commit comments