diff --git a/engines/tensorflow/tensorflow-api/build.gradle b/engines/tensorflow/tensorflow-api/build.gradle index b7c6c9934bc..d9d77e0329d 100644 --- a/engines/tensorflow/tensorflow-api/build.gradle +++ b/engines/tensorflow/tensorflow-api/build.gradle @@ -7,7 +7,7 @@ configurations { dependencies { api "org.bytedeco:javacpp:${javacpp_version}" api "com.google.protobuf:protobuf-java:${protobuf_version}" - api "org.tensorflow:tensorflow-core-api:0.4.2" + api "org.tensorflow:tensorflow-core-api:0.5.0" exclusion "org.bytedeco:javacpp:${javacpp_version}" exclusion "com.google.protobuf:protobuf-java:${protobuf_version}" @@ -19,6 +19,7 @@ jar { it.isDirectory() ? it : zipTree(it) } } + exclude "module-info.class" } publishing { diff --git a/engines/tensorflow/tensorflow-native/build.gradle b/engines/tensorflow/tensorflow-native/build.gradle index d72e52dc096..85a1670477f 100644 --- a/engines/tensorflow/tensorflow-native/build.gradle +++ b/engines/tensorflow/tensorflow-native/build.gradle @@ -13,11 +13,11 @@ configurations { } dependencies { - download "org.tensorflow:tensorflow-core-api:0.4.2:linux-x86_64-gpu" - download "org.tensorflow:tensorflow-core-api:0.4.2:linux-x86_64" - download "org.tensorflow:tensorflow-core-api:0.4.2:macosx-x86_64" - download "org.tensorflow:tensorflow-core-api:0.4.2:windows-x86_64" - download "org.tensorflow:tensorflow-core-api:0.4.2:windows-x86_64-gpu" + download "org.tensorflow:tensorflow-core-api:0.5.0:linux-x86_64-gpu" + download "org.tensorflow:tensorflow-core-api:0.5.0:linux-x86_64" + download "org.tensorflow:tensorflow-core-api:0.5.0:macosx-x86_64" + download "org.tensorflow:tensorflow-core-api:0.5.0:windows-x86_64" + download "org.tensorflow:tensorflow-core-api:0.5.0:windows-x86_64-gpu" } task uploadTensorflowNativeLibs() { @@ -76,6 +76,7 @@ task uploadTensorflowNativeLibs() { delete("${buildDir}/native/org/") delete("${buildDir}/native/com/") delete("${buildDir}/native/google/") + delete("${buildDir}/native/module-info.class.gz") exec { commandLine "aws", "s3", "sync", "${buildDir}/native/", "s3://djl-ai/publish/tensorflow-${tensorflow_version}/" diff --git a/gradle.properties b/gradle.properties index 81a09b7475e..dc82212e272 100644 --- a/gradle.properties +++ b/gradle.properties @@ -14,7 +14,7 @@ systemProp.org.gradle.internal.publish.checksums.insecure=true djl_version=0.22.0 mxnet_version=1.9.1 pytorch_version=1.13.1 -tensorflow_version=2.7.4 +tensorflow_version=2.10.1 tflite_version=2.6.2 trt_version=8.4.1 onnxruntime_version=1.14.0 @@ -36,7 +36,7 @@ slf4j_version=1.7.36 log4j_slf4j_version=2.19.0 awssdk_version=2.19.32 hadoop_version=3.3.4 -javacpp_version=1.5.6 +javacpp_version=1.5.8 protobuf_version=3.21.9 tablesaw_version=0.43.1 spark_version=3.2.2