Skip to content

Commit 3a51cbc

Browse files
HappyAmazoniansirutBuasai
authored andcommitted
[djl] neuronx patch releases for 0.28.0, 0.29.0,0.30.0 (aws#4489)
1 parent 1a2dca5 commit 3a51cbc

File tree

1 file changed

+8
-24
lines changed

1 file changed

+8
-24
lines changed

release_images_inference.yml

Lines changed: 8 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -125,11 +125,10 @@ release_images:
125125
version: "0.28.0"
126126
arch_type: "x86"
127127
inference:
128-
device_types: [ "gpu" ]
128+
device_types: [ "neuronx" ]
129129
python_versions: [ "py310" ]
130130
os_version: "ubuntu22.04"
131-
lmi_version: "10.0.0"
132-
cuda_version: "cu124"
131+
neuron_sdk_version: "sdk2.18.2"
133132
example: False
134133
disable_sm_tag: True
135134
force_release: False
@@ -138,11 +137,10 @@ release_images:
138137
version: "0.29.0"
139138
arch_type: "x86"
140139
inference:
141-
device_types: [ "gpu" ]
140+
device_types: [ "neuronx" ]
142141
python_versions: [ "py310" ]
143142
os_version: "ubuntu22.04"
144-
lmi_version: "11.0.0"
145-
cuda_version: "cu124"
143+
neuron_sdk_version: "sdk2.19.1"
146144
example: False
147145
disable_sm_tag: True
148146
force_release: False
@@ -151,28 +149,14 @@ release_images:
151149
version: "0.30.0"
152150
arch_type: "x86"
153151
inference:
154-
device_types: [ "gpu" ]
152+
device_types: [ "neuronx" ]
155153
python_versions: [ "py310" ]
156154
os_version: "ubuntu22.04"
157-
lmi_version: "12.0.0"
158-
cuda_version: "cu124"
155+
neuron_sdk_version: "sdk2.20.1"
159156
example: False
160157
disable_sm_tag: True
161158
force_release: False
162159
13:
163-
framework: "djl"
164-
version: "0.31.0"
165-
arch_type: "x86"
166-
inference:
167-
device_types: [ "gpu" ]
168-
python_versions: [ "py311" ]
169-
os_version: "ubuntu22.04"
170-
lmi_version: "13.0.0"
171-
cuda_version: "cu124"
172-
example: False
173-
disable_sm_tag: True
174-
force_release: False
175-
14:
176160
framework: "autogluon"
177161
version: "1.2.0"
178162
arch_type: "x86"
@@ -184,7 +168,7 @@ release_images:
184168
example: False
185169
disable_sm_tag: False
186170
force_release: False
187-
15:
171+
14:
188172
framework: "huggingface_pytorch"
189173
version: "2.3.0"
190174
arch_type: "x86"
@@ -197,7 +181,7 @@ release_images:
197181
example: False
198182
disable_sm_tag: True
199183
force_release: False
200-
16:
184+
15:
201185
framework: "huggingface_pytorch"
202186
version: "2.3.0"
203187
arch_type: "x86"

0 commit comments

Comments
 (0)