[onnxruntime] Updates OnnxRuntime engine to 1.19.0 (#3446)

deepjavalibrary · Aug 30, 2024 · d9a45ff · d9a45ff
1 parent 61d2f3a
commit d9a45ff
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git 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/...nes/onnxruntime/onnxruntime-engine/src/main/java/ai/djl/onnxruntime/engine/OrtEngine.java b/...nes/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
@@ -73,6 +73,7 @@ public void testOrt() throws TranslateException, ModelException, IOException {
             Criteria<IrisFlower, Classifications> 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
@@ -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"