From e7465fced86dce5d544f8c8f7c142ca7927c0978 Mon Sep 17 00:00:00 2001
From: Vincent Roseberry <rosbo@google.com>
Date: Fri, 30 Aug 2024 12:27:21 -0700
Subject: [PATCH] Add LiteRT
https://b.corp.google.com/issues/363294326
---
Dockerfile.tmpl | 2 ++
1 file changed, 2 insertions(+)
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index 8b3c3382..78306698 100644
--- 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 && \