[docker] Update DJL pytorch to 2.5.1 (#2555)

deepjavalibrary · Nov 14, 2024 · fa18ab2 · fa18ab2
1 parent b9e1c7d
commit fa18ab2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"