bump djl to 0.20 and add onnxruntime-gpu dependency (opensearch-proje…

…ct#644) Signed-off-by: Yaliang Wu <[email protected]> Signed-off-by: Yaliang Wu <[email protected]>
ylwu-amzn · Feb 17, 2023 · 37dd1f3 · 37dd1f3
1 parent 0a146a2
commit 37dd1f3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/ml-algorithms/build.gradle b/ml-algorithms/build.gradle
@@ -33,11 +33,14 @@ dependencies {
     testImplementation group: 'org.mockito', name: 'mockito-core', version: '4.4.0'
     testImplementation group: 'org.mockito', name: 'mockito-inline', version: '4.4.0'
     implementation group: 'com.google.guava', name: 'guava', version: '31.0.1-jre'
-    implementation platform("ai.djl:bom:0.19.0")
+    implementation platform("ai.djl:bom:0.20.0")
     implementation group: 'ai.djl.pytorch', name: 'pytorch-model-zoo'
     implementation group: 'ai.djl', name: 'api'
-    implementation group: 'ai.djl.onnxruntime', name: 'onnxruntime-engine'
     implementation group: 'ai.djl.huggingface', name: 'tokenizers'
+    implementation("ai.djl.onnxruntime:onnxruntime-engine:0.20.0") {
+        exclude group: "com.microsoft.onnxruntime", module: "onnxruntime"
+    }
+    implementation "com.microsoft.onnxruntime:onnxruntime_gpu:1.13.1"
 }
 
 configurations.all {

diff --git a/plugin/build.gradle b/plugin/build.gradle
@@ -47,7 +47,7 @@ dependencies {
     implementation("com.fasterxml.jackson.core:jackson-annotations:${versions.jackson}")
     implementation("com.fasterxml.jackson.core:jackson-databind:${versions.jackson_databind}")
     implementation group: 'com.google.guava', name: 'guava', version: '31.0.1-jre'
-    implementation group: 'com.google.code.gson', name: 'gson', version: '2.9.1'
+    implementation group: 'com.google.code.gson', name: 'gson', version: '2.10'
     implementation group: 'org.apache.commons', name: 'commons-lang3', version: '3.10'
     implementation group: 'org.apache.commons', name: 'commons-math3', version: '3.6.1'
     implementation group: 'org.apache.commons', name: 'commons-text', version: '1.10.0'