diff --git a/ml-algorithms/build.gradle b/ml-algorithms/build.gradle index 1516cae170..fa7c5054ae 100644 --- 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 index 2ab1ff1c65..d6065d0d00 100644 --- 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'