From 66e637a2ef37b2bd9101c1341767078373a29b66 Mon Sep 17 00:00:00 2001 From: Qing Lan Date: Wed, 12 Apr 2023 08:13:08 -0700 Subject: [PATCH 1/2] update container image --- src/sagemaker/image_uri_config/djl-deepspeed.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sagemaker/image_uri_config/djl-deepspeed.json b/src/sagemaker/image_uri_config/djl-deepspeed.json index 5e7cc8084d..70394d0d9a 100644 --- a/src/sagemaker/image_uri_config/djl-deepspeed.json +++ b/src/sagemaker/image_uri_config/djl-deepspeed.json @@ -29,7 +29,7 @@ "us-west-2": "763104351884" }, "repository": "djl-inference", - "tag_prefix": "0.21.0-deepspeed0.8.0-cu117" + "tag_prefix": "0.21.0-deepspeed0.8.3-cu117" }, "0.20.0": { "registries": { From 22978b82d706bb83386b49d43b19b8bf7dc8e0e4 Mon Sep 17 00:00:00 2001 From: Qing Lan Date: Wed, 12 Apr 2023 11:30:03 -0700 Subject: [PATCH 2/2] add test fix --- tests/unit/sagemaker/image_uris/test_djl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/sagemaker/image_uris/test_djl.py b/tests/unit/sagemaker/image_uris/test_djl.py index 4c15378ad1..9b5f8a2f77 100644 --- a/tests/unit/sagemaker/image_uris/test_djl.py +++ b/tests/unit/sagemaker/image_uris/test_djl.py @@ -47,7 +47,7 @@ "0.19.0": {"djl-deepspeed": "deepspeed0.7.3-cu113"}, "0.20.0": {"djl-deepspeed": "deepspeed0.7.5-cu116"}, "0.21.0": { - "djl-deepspeed": "deepspeed0.8.0-cu117", + "djl-deepspeed": "deepspeed0.8.3-cu117", "djl-fastertransformer": "fastertransformer5.3.0-cu117", }, }