From fa18ab2929025bdb334a9dfe61f2b957e0dde57a Mon Sep 17 00:00:00 2001
From: Xin Yang <105740670+xyang16@users.noreply.github.com>
Date: Thu, 14 Nov 2024 10:37:09 -0800
Subject: [PATCH] [docker] Update DJL pytorch to 2.5.1 (#2555)

---
 serving/docker/lmi.Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/serving/docker/lmi.Dockerfile b/serving/docker/lmi.Dockerfile
index 38fc86bc1..5ccab074f 100644
--- a/serving/docker/lmi.Dockerfile
+++ b/serving/docker/lmi.Dockerfile
@@ -18,7 +18,7 @@ ARG djl_serving_version
 ARG python_version=3.11
 ARG torch_version=2.5.1
 ARG torch_vision_version=0.20.1
-ARG djl_torch_version=2.4.0
+ARG djl_torch_version=2.5.1
 ARG onnx_version=1.19.0
 ARG pydantic_version=2.9.2
 ARG djl_converter_wheel="https://publish.djl.ai/djl_converter/djl_converter-0.31.0-py3-none-any.whl"