Skip to content

Commit

Permalink
[djl-bench] Add missing tensorflow model zoo dependency (#1143)
Browse files Browse the repository at this point in the history
Change-Id: I8d854c0d0353767cae81d999bf4f7fd9616ff5fc
  • Loading branch information
frankfliu authored Aug 4, 2021
1 parent 7b5bc04 commit e0aaa4b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions extensions/benchmark/build.gradle
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down

0 comments on commit e0aaa4b

Please sign in to comment.