feature: Bump up images for DJL transformers Neuronx DLCs (#3864)

siddvenk · web-flow · commit 0ecb38955727 · 2023-05-18T10:21:31.000-07:00
diff --git a/src/sagemaker/image_uri_config/djl-neuronx.json b/src/sagemaker/image_uri_config/djl-neuronx.json
@@ -29,7 +29,7 @@
                 "us-west-2": "763104351884"
             },
             "repository": "djl-inference",
-            "tag_prefix": "0.22.1-neuronx-sdk2.9.0"
+            "tag_prefix": "0.22.1-neuronx-sdk2.10.0"
         }
     }
 }
diff --git a/tests/unit/sagemaker/image_uris/test_djl.py b/tests/unit/sagemaker/image_uris/test_djl.py
@@ -54,7 +54,7 @@
     "0.22.1": {
         "djl-deepspeed": "deepspeed0.8.3-cu118",
         "djl-fastertransformer": "fastertransformer5.3.0-cu118",
-        "djl-neuronx": "neuronx-sdk2.9.0",
+        "djl-neuronx": "neuronx-sdk2.10.0",
     },
 }
 

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,7 @@`
`54`	`54`	`"0.22.1": {`
`55`	`55`	`"djl-deepspeed": "deepspeed0.8.3-cu118",`
`56`	`56`	`"djl-fastertransformer": "fastertransformer5.3.0-cu118",`
`57`		`- "djl-neuronx": "neuronx-sdk2.9.0",`
	`57`	`+ "djl-neuronx": "neuronx-sdk2.10.0",`
`58`	`58`	`},`
`59`	`59`	`}`
`60`	`60`