From 59539b634e51b5e543fa6a8bb77da9069df15eae Mon Sep 17 00:00:00 2001 From: pintaoz Date: Tue, 13 May 2025 12:30:17 -0700 Subject: [PATCH 1/2] Fix test_huggingface_tei_uris() --- tests/unit/sagemaker/image_uris/test_huggingface_llm.py | 2 ++ 1 file changed, 2 insertions(+) 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 = { From d8ea4e8247d72b0437ac850fbd5eb8d77692394f Mon Sep 17 00:00:00 2001 From: pintaoz Date: Tue, 13 May 2025 13:44:53 -0700 Subject: [PATCH 2/2] Fix json --- src/sagemaker/image_uri_config/huggingface-llm-neuronx.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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": {