Skip to content

Commit 849ed29

Browse files
authored
change: Add djl 0.21.0 dlc images (#3685)
1 parent 1ad9d9c commit 849ed29

File tree

2 files changed

+32
-1
lines changed

2 files changed

+32
-1
lines changed

src/sagemaker/image_uri_config/djl-deepspeed.json

+30
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,36 @@
11
{
22
"scope": ["inference"],
33
"versions": {
4+
"0.21.0": {
5+
"registries": {
6+
"af-south-1": "626614931356",
7+
"ap-east-1": "871362719292",
8+
"ap-northeast-1": "763104351884",
9+
"ap-northeast-2": "763104351884",
10+
"ap-northeast-3": "364406365360",
11+
"ap-south-1": "763104351884",
12+
"ap-southeast-1": "763104351884",
13+
"ap-southeast-2": "763104351884",
14+
"ap-southeast-3": "907027046896",
15+
"ca-central-1": "763104351884",
16+
"cn-north-1": "727897471807",
17+
"cn-northwest-1": "727897471807",
18+
"eu-central-1": "763104351884",
19+
"eu-north-1": "763104351884",
20+
"eu-west-1": "763104351884",
21+
"eu-west-2": "763104351884",
22+
"eu-west-3": "763104351884",
23+
"eu-south-1": "692866216735",
24+
"me-south-1": "217643126080",
25+
"sa-east-1": "763104351884",
26+
"us-east-1": "763104351884",
27+
"us-east-2": "763104351884",
28+
"us-west-1": "763104351884",
29+
"us-west-2": "763104351884"
30+
},
31+
"repository": "djl-inference",
32+
"tag_prefix": "0.21.0-deepspeed0.8.0-cu117"
33+
},
434
"0.20.0": {
535
"registries": {
636
"af-south-1": "626614931356",

tests/unit/sagemaker/image_uris/test_djl.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -41,11 +41,12 @@
4141
"us-west-1": "763104351884",
4242
"us-west-2": "763104351884",
4343
}
44-
VERSIONS = ["0.20.0", "0.19.0"]
44+
VERSIONS = ["0.21.0", "0.20.0", "0.19.0"]
4545
DJL_FRAMEWORKS = ["djl-deepspeed"]
4646
DJL_VERSIONS_TO_FRAMEWORK = {
4747
"0.19.0": {"djl-deepspeed": "deepspeed0.7.3-cu113"},
4848
"0.20.0": {"djl-deepspeed": "deepspeed0.7.5-cu116"},
49+
"0.21.0": {"djl-deepspeed": "deepspeed0.8.0-cu117"},
4950
}
5051

5152

0 commit comments

Comments
 (0)