[djl-bench] Add missing tensorflow model zoo dependency (#1143)

Change-Id: I8d854c0d0353767cae81d999bf4f7fd9616ff5fc
deepjavalibrary · Aug 4, 2021 · e0aaa4b · e0aaa4b
1 parent 7b5bc04
commit e0aaa4b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/extensions/benchmark/build.gradle b/extensions/benchmark/build.gradle
@@ -37,6 +37,7 @@ dependencies {
         runtimeOnly project(":pytorch:pytorch-model-zoo")
         runtimeOnly "ai.djl.pytorch:pytorch-native-auto:${pytorch_version}-SNAPSHOT"
 
+        runtimeOnly project(":tensorflow:tensorflow-model-zoo")
         runtimeOnly "ai.djl.tensorflow:tensorflow-native-auto:${tensorflow_version}"
 
         runtimeOnly project(":mxnet:mxnet-model-zoo")