Skip to content

Commit 9dbead8

Browse files
authored
change: update LMI container image (#3782)
* update container image * add test fix
1 parent 4c1c118 commit 9dbead8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sagemaker/image_uri_config/djl-deepspeed.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
"us-west-2": "763104351884"
3030
},
3131
"repository": "djl-inference",
32-
"tag_prefix": "0.21.0-deepspeed0.8.0-cu117"
32+
"tag_prefix": "0.21.0-deepspeed0.8.3-cu117"
3333
},
3434
"0.20.0": {
3535
"registries": {

tests/unit/sagemaker/image_uris/test_djl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@
4747
"0.19.0": {"djl-deepspeed": "deepspeed0.7.3-cu113"},
4848
"0.20.0": {"djl-deepspeed": "deepspeed0.7.5-cu116"},
4949
"0.21.0": {
50-
"djl-deepspeed": "deepspeed0.8.0-cu117",
50+
"djl-deepspeed": "deepspeed0.8.3-cu117",
5151
"djl-fastertransformer": "fastertransformer5.3.0-cu117",
5252
},
5353
}

0 commit comments

Comments
 (0)