change: Upgrade DJL deepspeed versions (#3921)

Qing Lan · web-flow · commit a8b6b1dbbe2d · 2023-07-06T08:28:09.000-07:00
diff --git a/src/sagemaker/image_uri_config/djl-deepspeed.json b/src/sagemaker/image_uri_config/djl-deepspeed.json
@@ -30,7 +30,7 @@
                 "us-west-2": "763104351884"
             },
             "repository": "djl-inference",
-            "tag_prefix": "0.22.1-deepspeed0.8.3-cu118"
+            "tag_prefix": "0.22.1-deepspeed0.9.2-cu118"
         },
         "0.21.0": {
             "registries": {
diff --git a/tests/unit/sagemaker/image_uris/test_djl.py b/tests/unit/sagemaker/image_uris/test_djl.py
@@ -53,7 +53,7 @@
         "djl-fastertransformer": "fastertransformer5.3.0-cu117",
     },
     "0.22.1": {
-        "djl-deepspeed": "deepspeed0.8.3-cu118",
+        "djl-deepspeed": "deepspeed0.9.2-cu118",
         "djl-fastertransformer": "fastertransformer5.3.0-cu118",
         "djl-neuronx": "neuronx-sdk2.10.0",
     },
diff --git a/tests/unit/sagemaker/image_uris/test_huggingface_llm.py b/tests/unit/sagemaker/image_uris/test_huggingface_llm.py
@@ -50,7 +50,7 @@
     "0.6.0": "2.0.0-tgi0.6.0-gpu-py39-cu118-ubuntu20.04",
     "0.8.2": "2.0.0-tgi0.8.2-gpu-py39-cu118-ubuntu20.04",
 }
-LMI_VERSIONS_MAPPING = {"0.22.1": "deepspeed0.8.3-cu118"}
+LMI_VERSIONS_MAPPING = {"0.22.1": "deepspeed0.9.2-cu118"}
 
 
 @pytest.mark.parametrize("version", HF_VERSIONS)

Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@`
`50`	`50`	`"0.6.0": "2.0.0-tgi0.6.0-gpu-py39-cu118-ubuntu20.04",`
`51`	`51`	`"0.8.2": "2.0.0-tgi0.8.2-gpu-py39-cu118-ubuntu20.04",`
`52`	`52`	`}`
`53`		`-LMI_VERSIONS_MAPPING = {"0.22.1": "deepspeed0.8.3-cu118"}`
	`53`	`+LMI_VERSIONS_MAPPING = {"0.22.1": "deepspeed0.9.2-cu118"}`
`54`	`54`
`55`	`55`
`56`	`56`	`@pytest.mark.parametrize("version", HF_VERSIONS)`