diff --git a/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json b/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json index 74647b107a..9b7b18ee94 100644 --- a/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json +++ b/src/sagemaker/image_uri_config/huggingface-llm-neuronx.json @@ -4,7 +4,7 @@ "inf2" ], "version_aliases": { - "0.0": "0.0.28", + "0.0": "0.0.28" }, "versions": { "0.0.16": { diff --git a/tests/unit/sagemaker/image_uris/test_huggingface_llm.py b/tests/unit/sagemaker/image_uris/test_huggingface_llm.py index 6598117027..e693b9f8ce 100644 --- a/tests/unit/sagemaker/image_uris/test_huggingface_llm.py +++ b/tests/unit/sagemaker/image_uris/test_huggingface_llm.py @@ -24,11 +24,13 @@ "1.2.3": "2.0.1-tei1.2.3-gpu-py310-cu122-ubuntu22.04", "1.4.0": "2.0.1-tei1.4.0-gpu-py310-cu122-ubuntu22.04", "1.6.0": "2.0.1-tei1.6.0-gpu-py310-cu122-ubuntu22.04", + "1.7.0": "2.0.1-tei1.7.0-gpu-py310-cu122-ubuntu22.04", }, "cpu": { "1.2.3": "2.0.1-tei1.2.3-cpu-py310-ubuntu22.04", "1.4.0": "2.0.1-tei1.4.0-cpu-py310-ubuntu22.04", "1.6.0": "2.0.1-tei1.6.0-cpu-py310-ubuntu22.04", + "1.7.0": "2.0.1-tei1.7.0-cpu-py310-ubuntu22.04", }, } HF_VERSIONS_MAPPING = {