Add dependency on ai-edge-litert-nightly for ai-edge-torch pypi.

PiperOrigin-RevId: 677876188
google-ai-edge · Sep 23, 2024 · 6dcb213 · 6dcb213
1 parent 0233ab9
commit 6dcb213
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/ai_edge_torch/generative/examples/phi/verify.py b/ai_edge_torch/generative/examples/phi/verify.py
@@ -34,6 +34,7 @@
     "The maximum size of the generated tokens.",
 )
 
+
 def main(_):
   checkpoint = kagglehub.model_download("Microsoft/phi/transformers/2")
   verifier.log_msg("Loading the original model from", checkpoint)

diff --git a/setup.py b/setup.py
@@ -88,6 +88,7 @@
         "torch>=2.4.0",
         "torch_xla>=2.4.0",
         "tf-nightly>=2.18.0.dev20240722",
+        "ai-edge-litert-nightly",
         "ai-edge-quantizer-nightly",
     ],
 )