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": { 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", }, }