diff --git a/release_images_inference.yml b/release_images_inference.yml
index b3ee57473c3b..80f94f413ae8 100644
--- a/release_images_inference.yml
+++ b/release_images_inference.yml
@@ -58,13 +58,14 @@ release_images:
       force_release: False
   5:
     framework: "djl"
-    version: "0.30.0"
+    version: "0.31.0"
     arch_type: "x86"
     inference:
-      device_types: [ "neuronx" ]
-      python_versions: [ "py310" ]
+      device_types: [ "gpu" ]
+      python_versions: [ "py311" ]
       os_version: "ubuntu22.04"
-      neuron_sdk_version: "sdk2.20.1"
+      lmi_version: "13.0.0"
+      cuda_version: "cu124"
       example: False
       disable_sm_tag: True
       force_release: True