diff --git a/oss_scripts/pip_package/requirements.in b/oss_scripts/pip_package/requirements.in
index 54540bf47..147a2cb8f 100644
--- a/oss_scripts/pip_package/requirements.in
+++ b/oss_scripts/pip_package/requirements.in
@@ -1,4 +1,5 @@
+ai-edge-litert
 tensorflow
 tf-keras
 tensorflow-datasets
-tensorflow-metadata
+tensorflow-metadata
\ No newline at end of file