Skip to content

Commit 962c394

Browse files
authored
change: DJL Neuronx 0.24.0 (#4170)
1 parent ea60a2d commit 962c394

File tree

2 files changed

+35
-0
lines changed

2 files changed

+35
-0
lines changed

src/sagemaker/image_uri_config/djl-neuronx.json

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,37 @@
11
{
22
"scope": ["inference"],
33
"versions": {
4+
"0.24.0": {
5+
"registries": {
6+
"af-south-1": "626614931356",
7+
"il-central-1": "780543022126",
8+
"ap-east-1": "871362719292",
9+
"ap-northeast-1": "763104351884",
10+
"ap-northeast-2": "763104351884",
11+
"ap-northeast-3": "364406365360",
12+
"ap-south-1": "763104351884",
13+
"ap-southeast-1": "763104351884",
14+
"ap-southeast-2": "763104351884",
15+
"ap-southeast-3": "907027046896",
16+
"ca-central-1": "763104351884",
17+
"cn-north-1": "727897471807",
18+
"cn-northwest-1": "727897471807",
19+
"eu-central-1": "763104351884",
20+
"eu-north-1": "763104351884",
21+
"eu-west-1": "763104351884",
22+
"eu-west-2": "763104351884",
23+
"eu-west-3": "763104351884",
24+
"eu-south-1": "692866216735",
25+
"me-south-1": "217643126080",
26+
"sa-east-1": "763104351884",
27+
"us-east-1": "763104351884",
28+
"us-east-2": "763104351884",
29+
"us-west-1": "763104351884",
30+
"us-west-2": "763104351884"
31+
},
32+
"repository": "djl-inference",
33+
"tag_prefix": "0.24.0-neuronx-sdk2.14.1"
34+
},
435
"0.23.0": {
536
"registries": {
637
"af-south-1": "626614931356",

tests/unit/sagemaker/image_uris/test_djl.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
DJL_DEEPSPEED_VERSIONS = ["0.23.0", "0.22.1", "0.21.0", "0.20.0", "0.19.0"]
4646
DJL_FASTERTRANSFORMER_VERSIONS = ["0.23.0", "0.22.1", "0.21.0"]
4747
DJL_NEURONX_VERSIONS = ["0.23.0", "0.22.1"]
48+
DJL_NEURONX_VERSIONS = ["0.24.0", "0.23.0", "0.22.1"]
4849
DJL_VERSIONS_TO_FRAMEWORK = {
4950
"0.19.0": {"djl-deepspeed": "deepspeed0.7.3-cu113"},
5051
"0.20.0": {"djl-deepspeed": "deepspeed0.7.5-cu116"},
@@ -62,6 +63,9 @@
6263
"djl-fastertransformer": "fastertransformer5.3.0-cu118",
6364
"djl-neuronx": "neuronx-sdk2.12.0",
6465
},
66+
"0.24.0": {
67+
"djl-neuronx": "neuronx-sdk2.14.1",
68+
},
6569
}
6670

6771

0 commit comments

Comments
 (0)