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 && \