Add LiteRT

https://b.corp.google.com/issues/363294326
Kaggle · Aug 30, 2024 · e7465fc · e7465fc
1 parent 9e84258
commit e7465fc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
@@ -205,6 +205,8 @@ RUN pip install \
         tensorflow_decision_forests \
         tensorflow-text \
         "tensorflow_hub>=0.16.0" \
+	# LiteRT (previously known as TFLite).
+	ai_edge_litert \
         # b/331799280 remove once other packages over to dm-tre
         optree \
         tf-keras && \