diff --git a/bom/build.gradle.kts b/bom/build.gradle.kts index 9c4e87c58cc..3ee65ff6df1 100644 --- a/bom/build.gradle.kts +++ b/bom/build.gradle.kts @@ -48,6 +48,7 @@ dependencies { api("ai.djl.timeseries:timeseries:${version}") api("com.microsoft.onnxruntime:onnxruntime:${libs.versions.onnxruntime.get()}") api("com.microsoft.onnxruntime:onnxruntime_gpu:${libs.versions.onnxruntime.get()}") + api("com.microsoft.onnxruntime:onnxruntime-extensions:${libs.versions.onnxruntimeExtensions.get()}") } } diff --git a/engines/onnxruntime/onnxruntime-engine/src/main/java/ai/djl/onnxruntime/engine/OrtEngine.java b/engines/onnxruntime/onnxruntime-engine/src/main/java/ai/djl/onnxruntime/engine/OrtEngine.java index ee843bfdd04..c3c2ea0eea0 100644 --- a/engines/onnxruntime/onnxruntime-engine/src/main/java/ai/djl/onnxruntime/engine/OrtEngine.java +++ b/engines/onnxruntime/onnxruntime-engine/src/main/java/ai/djl/onnxruntime/engine/OrtEngine.java @@ -102,7 +102,7 @@ public int getRank() { /** {@inheritDoc} */ @Override public String getVersion() { - return "1.18.0"; + return "1.19.0"; } /** {@inheritDoc} */ diff --git a/engines/onnxruntime/onnxruntime-engine/src/test/java/ai/djl/onnxruntime/engine/OrtTest.java b/engines/onnxruntime/onnxruntime-engine/src/test/java/ai/djl/onnxruntime/engine/OrtTest.java index dce45197558..df6fa99fcb9 100644 --- a/engines/onnxruntime/onnxruntime-engine/src/test/java/ai/djl/onnxruntime/engine/OrtTest.java +++ b/engines/onnxruntime/onnxruntime-engine/src/test/java/ai/djl/onnxruntime/engine/OrtTest.java @@ -73,6 +73,7 @@ public void testOrt() throws TranslateException, ModelException, IOException { Criteria criteria = Criteria.builder() .setTypes(IrisFlower.class, Classifications.class) + .optModelUrls("djl://ai.djl.onnxruntime/iris_flowers") .optEngine("OnnxRuntime") // use OnnxRuntime engine .optOption("interOpNumThreads", "1") .optOption("intraOpNumThreads", "1") diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml index ce3a61619d3..7ac16b108d4 100644 --- a/gradle/libs.versions.toml +++ b/gradle/libs.versions.toml @@ -7,8 +7,8 @@ pytorch = "2.4.0" tensorflow = "2.16.1" tensorflowCore = "1.0.0-rc.1" mxnet = "1.9.1" -onnxruntime = "1.18.0" -onnxruntimeExtensions = "0.11.0" +onnxruntime = "1.19.0" +onnxruntimeExtensions = "0.12.4" xgboost = "2.0.3" lightgbm = "3.2.110" tensorrt = "8.4.1"