upgrade deepspeed versions

Qing Lan · Qing Lan · commit 40a9ef705382 · 2023-06-28T08:30:15.000-07:00
diff --git 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.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
@@ -52,7 +52,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
@@ -49,7 +49,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
`@@ -49,7 +49,7 @@`
`49`	`49`	`"0.6.0": "2.0.0-tgi0.6.0-gpu-py39-cu118-ubuntu20.04",`
`50`	`50`	`"0.8.2": "2.0.0-tgi0.8.2-gpu-py39-cu118-ubuntu20.04",`
`51`	`51`	`}`
`52`		`-LMI_VERSIONS_MAPPING = {"0.22.1": "deepspeed0.8.3-cu118"}`
	`52`	`+LMI_VERSIONS_MAPPING = {"0.22.1": "deepspeed0.9.2-cu118"}`
`53`	`53`
`54`	`54`
`55`	`55`	`@pytest.mark.parametrize("version", HF_VERSIONS)`