Skip to content

Commit a7f1cab

Browse files
authored
fix: image uri in TGI 2.2.0 image (aws#4804)
1 parent 3b99806 commit a7f1cab

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sagemaker/image_uri_config/huggingface-llm.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -717,7 +717,7 @@
717717
"tag_prefix": "2.3.0-tgi2.2.0",
718718
"repository": "huggingface-pytorch-tgi-inference",
719719
"container_version": {
720-
"gpu": "cu121-ubuntu22.04"
720+
"gpu": "cu121-ubuntu22.04-v2.0"
721721
}
722722
}
723723
}

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@
4242
"2.0.0": "2.1.1-tgi2.0.0-gpu-py310-cu121-ubuntu22.04",
4343
"2.0.1": "2.1.1-tgi2.0.1-gpu-py310-cu121-ubuntu22.04",
4444
"2.0.2": "2.3.0-tgi2.0.2-gpu-py310-cu121-ubuntu22.04",
45-
"2.2.0": "2.3.0-tgi2.2.0-gpu-py310-cu121-ubuntu22.04",
45+
"2.2.0": "2.3.0-tgi2.2.0-gpu-py310-cu121-ubuntu22.04-v2.0",
4646
},
4747
"inf2": {
4848
"0.0.16": "1.13.1-optimum0.0.16-neuronx-py310-ubuntu22.04",

0 commit comments

Comments
 (0)