google-ai-edge/LiteRT-CLI

google-ai-edge 资料

最近记录的来源活动: 2026年9月29日 04:15

SkillsMP 收录数据更新: 2026年10月17日 00:09

已收集 skills: 1

GitHub 星标: 43
GitHub Forks: 8

仓库 README

仓库 README 来源：README.md

查看 README 源文件

展开完整 README

# LiteRT CLI (Preview) A convenient command-line toolkit to streamline [LiteRT](https://ai.google.dev/edge/litert) related development workflows, including converting, quantizing, compiling, running, benchmarking and visualizing LiteRT (TFLite) models on various hardware (CPU / GPU / NPU) across platforms (desktop, mobile, or cloud). 🚀 [Installation](#-installation) ｜ 📓 [Try Colab](https://github.com/google-ai-edge/LiteRT-CLI/blob/main/examples/litert_cli.ipynb) | ⚡ [LiteRT 101](https://codelabs.developers.google.com/codelabs/litert-cli/101) | ⚡ [Quick start](#-quick-start) | 💡 [Commands](#-common-commands) | 🤖 [Coding agent](#-use-in-coding-agent) > [!NOTE] > > It's still an early preview under active development, thus has limited > platform and feature support, plus possible bugs. We appreciate your patience > and feedback to help us improve it. Welcome issues and PRs! LiteRT CLI is built on top of [Google AI Edge](https://ai.google.dev/edge) stacks, including [LiteRT](https://github.com/google-ai-edge/LiteRT), [LiteRT-LM](https://github.com/google-ai-edge/LiteRT-LM), [LiteRT Torch](https://github.com/google-ai-edge/LiteRT-Torch), [LiteRT Quantizer](https://github.com/google-ai-edge/ai-edge-quantizer), [AI Edge Portal](https://ai.google.dev/edge/ai-edge-portal), and [Model Explorer](https://ai.google.dev/edge/model-explorer). -------------------------------------------------------------------------------- ## ⚙️ Build Status Build Type | Status ------------------------- | ------ Unit Tests (Linux, macOS) | [![](https://github.com/google-ai-edge/LiteRT-CLI/actions/workflows/nightly_unittests.yml/badge.svg?branch=main)](https://github.com/google-ai-edge/LiteRT-CLI/actions/workflows/nightly_unittests.yml) Nightly Release | [![](https://github.com/google-ai-edge/LiteRT-CLI/actions/workflows/nightly_release.yml/badge.svg?branch=main)](https://github.com/google-ai-edge/LiteRT-CLI/actions/workflows/nightly_release.yml) -------------------------------------------------------------------------------- ## 🚀 Installation Please install `litert-cli-nightly` from PyPI or from local clone. LiteRT CLI will install the dependencies on-demand, based on which commands to run, to speed up initial installation. We support installation using either **[uv](https://docs.astral.sh/uv/getting-started/installation/)** (recommended for ultra-fast dependency resolution) or standard **[pip](https://pip.pypa.io/)** within a Python virtual environment. ### Option 1: Use uv (recommended) `uv` is an extremely fast Python package manager written in Rust. ```bash # 1. Create a virtual environment with Python 3.13. # TIP: Sometimes setting env var `UV_INDEX_URL=https://pypi.org/simple` helps # resolve dependency resolution errors, especially inside corporate networks. uv venv --clear --python=3.13 --seed source .venv/bin/activate # 2. Install the package into the active virtual environment uv pip install litert-cli-nightly # 3. Run help command litert --help ``` ### Option 2: Use standard pip ```bash python3 -m venv .venv source .venv/bin/activate pip install -q litert-cli-nightly litert --help ``` ### Option 3: Install from local clone (for development) ```bash uv venv --clear --python=3.13 --seed source .venv/bin/activate git clone git@github.com:google-ai-edge/LiteRT-CLI.git cd LiteRT-CLI uv pip install -e . ``` -------------------------------------------------------------------------------- ## ⚡ Quick start ### 📓 Try Colab Try [LiteRT CLI Colab](https://github.com/google-ai-edge/LiteRT-CLI/blob/main/examples/litert_cli.ipynb) to explore different features quickly. ### 📓 LiteRT 101 [LiteRT 101](https://codelabs.developers.google.com/codelabs/litert-cli/101) is a hands-on tutorial that walks you through the basics of using the LiteRT CLI tool. ### Follow command help You can always follow `litert --help` or `litert {command} --help` to find how to use the CLI tool. Check [detailed instructions for each command](#-common-commands) below. ```bash # Run help command litert --help # Download a LiteRT model litert download --help litert download litert-community/efficientnet_b1 --file "*.tflite" --output efficientnet # Run and benchmark a LiteRT model on your devices litert run --help litert run efficientnet/efficientnet_b1.tflite --desktop --cpu litert benchmark --help litert benchmark efficientnet/efficientnet_b1.tflite --android --gpu ``` ### Run LLMs You can run LLMs like Gemma4-E2B, using commands like `litert lm run`, which is an alias to `litert-lm run`, and please follow [LiteRT-LM CLI](https://developers.google.com/edge/litert-lm/cli) for detailed instructions. ```bash litert lm run \ --from-huggingface-repo=litert-community/gemma-4-E2B-it-litert-lm \ gemma-4-E2B-it.litertlm \ --prompt="What is the capital of France?" ``` ### 🌟 Quick demos Check comprehensive usage examples under the [examples/](https://github.com/google-ai-edge/LiteRT-CLI/tree/main/examples) directory, which contains per-command demos and model-specific demos. If you have cloned the repo, you can run the following commands to see the demos. Note: running all demos will take time and disk space. ```bash # Run all command demos ./examples/run_commands.sh # Run specific command demos ./examples/run_commands.sh download,benchmark # Run all model demos ./examples/run_models.sh # Run a specific model demo ./examples/run_models.sh efficientnet ``` ### 🤖 Use in coding agent Add the LiteRT CLI skill [`SKILL.md`](https://github.com/google-ai-edge/LiteRT-CLI/blob/main/.agents/skills/litert_cli/SKILL.md) into your coding agent (like [Google Antigravity](https://antigravity.google/)) and try prompts such as: * *Download LiteRT model `litert-community/efficientnet_b1` and run it on CPU* * *Benchmark LiteRT model `litert-community/efficientnet_b1` on my Android GPU* * *Compile LiteRT model `litert-community/efficientnet_b1` for NPU target `sm8750`* * *Visualize LiteRT model `litert-community/efficientnet_b1`* * *Download the FP32 model `litert-community/efficientnet_b1` , quantize it to INT8 dynamic range (`--recipe dynamic_wi8_afp32`), then benchmark both the original FP32 model and the newly quantized INT8 model on the GPU of my connected Android device. Compare the average latency and report the throughput speedup.* * *Convert the model `Qwen/Qwen1.5-0.5B-Chat` from HuggingFace, and run it locally using the prompt 'Explain edge machine learning one sentence'* * *Download EfficientNet from huggingface repo `litert-community/efficientnet_b1`, offline compile (AOT) the model for the `sm8750` target NPU, and output the compiled model into `./models/compiled`. Then, run an on-device inference and benchmark using this newly compiled AOT model on the connected Android device's NPU (`--npu`). Confirm that the graph loads directly without dynamic JIT compilation warmup latency.* The agent will automatically install the necessary tools, including Python virtual environments, `litert-cli-nightly`, and all required dependencies. -------------------------------------------------------------------------------- ### Verified platforms Verified in Python 3.13. * **Host Machines**: * Linux (Ubuntu) * macOS (Apple Silicon): don't support `litert compile` yet. * Windows: `litert compile` and `litert convert` not supported yet. * **Android**: * CPU, GPU * NPU: Qualcomm, MediaTek (soon), Google Tensor (soon) -------------------------------------------------------------------------------- ### Troubleshooting and tips * Always activate python virtual environment before running `litert` command, to avoid conflicts. * When `uv` fails to resolve dependencies, try to set below environment variable first: `export UV_INDEX_URL=https://pypi.org/simple`. * When run fails on GPU using `--gpu` flag, try to add both `--cpu --gpu` flags in the command, then the CLI will try CPU first, and fall back to GPU when CPU fails. * When `litert run` fails on Android device, if the device is not detected, try to run `adb kill-server` first. * When convert a traditional PyTorch model, you need to write a script to wrap it with required functions `get_model` and `get_args`. Check the script format in [resnet18.py](https://github.com/google-ai-edge/LiteRT-CLI/blob/main/litert_cli/test_data/resnet18.py). * LLM conversion only supports HuggingFace models with type AutoModelForCausalLM and * Gemma family now. * For large models like LLMs, `litert convert` will take large memories and disks, and spend multiple minutes. Please make sure you have enough memory and disks, and be patient. * `litert compile` only supports running on Linux now, and it requires newer Clang version `18.x.x` or above. Try `sudo apt install clang libc++-18-dev libc++abi-18-dev`. * When benchmark using `--gcp` flag, you need to 1) Join the EAP program of [Google AI Edge Portal](https://ai.google.dev/edge/ai-edge-portal); 2) Login to GCP using `gcloud auth login`; 3) Set your GCP project using `--gcp=<Your-GCP-Project>`. * When benchmark using `--ddp` flag, you need to 1) Enable the Device Run API of [Developer Device Platform](https://docs.cloud.google.com/developer-device-platform/overview) in a GCP project with billing enabled (the platform is in Preview, and sessions are billed to your project); 2) Login using `gcloud auth application-default login`; 3) Set your GCP project using `--gcp-project=<Your-GCP-Project>`. The devices run the prebuilt `benchmark_model` of LiteRT 2.2.0; set the environment variable `DDP_LITERT_VERSION` (for example `nightly`) to pick another release. A `.litertlm` bundle runs LiteRT-LM's prebuilt benchmark binary from `gs://litert/binaries/latest/android_arm64/litert_lm/`, with the shared libraries of that directory pushed beside it; `DDP_LITERT_LM_VERSION` picks another version there, and `DDP_LITERT_LM_DIR` a whole `gs://` directory laid out the same way. * When `litert visualize` fails to launch Model Explorer, try to run `litert visualize --stop-all` first. * Exporting environment variable `LITERT_VERBOSE=1` can enable verbose logging. * `litert clean` will clean all local caches, like model files and binaries, which will free your disk space, and further, it will be very helpful for fixing complicated issues, like issues caused by NPU library version mismatch. -------------------------------------------------------------------------------- ## 💡 Common commands ### 1. Download a model from Hugging Face Hub ```bash # Download only .tflite files litert download litert-community/MobileNet-v3-large \ --file "*.tflite" \ --output mobilenet # Download full repository litert download litert-community/MobileNet-v3-large \ --output mobilenet_full # Download models using Hugging Face ID (uses HF ID as model reference too) litert download litert-community/MobileNet-v3-large # Download models with custom model reference litert download litert-community/MobileNet-v3-large --model-ref my_model_ref ``` ### 2. Convert a PyTorch model into a LiteRT model ```bash # Automated HF Conversion litert convert Qwen/Qwen1.5-0.5B-Chat --output /tmp/qwen # Automated HF Conversion with INT4 Weight-Only Quantization litert convert Qwen/Qwen1.5-0.5B-Chat --quantize-recipe weight_only_wi4_afp32 --output /tmp/qwen_w4 # Generic Script Injection with INT8 Dynamic Quantization litert convert my_model.py --quantize-recipe dynamic_wi8_afp32 --output /tmp/mymodel # Use a custom LlmMetadata (chat template, stop tokens) for the .litertlm bundle litert convert Qwen/Qwen3-1.7B --llm-metadata my_metadata.pbtext --output /tmp/qwen3 ``` For models with a canonical config in [LiteRT-LM `models/<family>`](https://github.com/google-ai-edge/LiteRT-LM/tree/main/models), such as Qwen2.5, Qwen3, Qwen3.5, LFM2 and MiniCPM5, `litert convert` tries to reuse that config. ### 3. Quantize a LiteRT model ```bash # Dynamic INT8 Quantization (Default) litert quantize model.tflite \ --recipe dynamic_wi8_afp32 \ --output dynamic.tflite # Weight-Only Quantization litert quantize model.tflite \ --recipe weight_only_wi8_afp32 \ --output weight_only.tflite # Static W8A8 Quantization (with calibration data) litert quantize model.tflite \ --recipe static_wi8_ai8 \ --calibration-data calib_data.py \ --output static.tflite # Custom Recipe litert quantize model.tflite \ --custom-recipe quantize_recipe.json \ --output custom_quant.tflite ``` ### 4. AOT Compile a LiteRT model for NPU > [!NOTE] > > Currently only support on Linux hosts and Qualcomm NPUs, and other NPU > supports are coming soon! ```bash # Basic compilation for specific Qualcomm NPU (e.g., sm8750) litert compile model.tflite --target sm8750 # Compile for multiple targets and export an AI Pack for Android litert compile model.tflite --target sm8750 --target mt6989 --export-aipack my_npu_models ``` ### 5. Run a LiteRT model on desktop or Android ```bash # Run locally on desktop (CPU) litert run model.tflite --desktop --cpu litert run my_model_ref --desktop --cpu # Run with GPU acceleration and CPU fallback (multi-accelerator) litert run model.tflite --gpu --cpu litert run model.tflite --accelerator gpu,cpu # Run on connected Android device litert run model.tflite --android # Run on connected Android device with NPU acceleration and CPU fallback litert run model.tflite --android --npu --cpu litert run model.tflite --android --accelerator npu,cpu # Run on connected Android device with NPU AOT-compiled model litert run model_sm8450.tflite --android --npu # Run multiple iterations and print output tensors litert run model.tflite \ --iterations 5 \ --print-tensors # Run with custom input formats (supports image, raw binary, numpy array) litert run model.tflite \ --input "image.png" \ --print-tensors ``` ### 6. Benchmark a LiteRT model ```bash # On the CPU and the GPU (--android, --desktop, --ddp), benchmark_model runs twice: a first process # compiles the model and writes the XNNPACK and GPU caches (no inference), then the measured process # reads them and reports the peak memory too. The GPU cache needs a benchmark_model with # --gpu_serialization_dir: the nightly binaries have it, 2.2.0 does not. If a process with the cache # flags fails, benchmark_model runs once without them, as before, and the report says so; # LITERT_DISABLE_MODEL_CACHES=1 runs it once from the start. # Benchmark on Android (CPU side) litert benchmark my_model_ref --android --cpu litert benchmark model.tflite --android --cpu # Benchmark on Android NPU (JIT mode) litert benchmark model.tflite --android --npu # Benchmark AOT compiled model on Android NPU litert benchmark model_sm8450.tflite --android --npu # Benchmark on Android GPU litert benchmark model.tflite --android --gpu # Benchmark on macOS (CPU) litert benchmark my_model_ref --desktop --cpu # Benchmark on Google AI Edge Portal in Google Cloud. Prerequisites: # - Set up your Google AI Edge Portal account by following up the instructions at: # https://ai.google.dev/edge/ai-edge-portal # - Set up authentication by running: gcloud auth login # - You can set the default GCP project by setting the environment variable LITERT_GCP_PROJECT, or by providing the --gcp-project option. # - You can specific your GCP bucket by --gcp-bucket, otherwise, it will create default # one. litert benchmark model.tflite --gcp --device "pixel 7" --gcp-project "your-gcp-project-id" --gcp-bucket "your-gcp-bucket" litert benchmark model.tflite --gcp --devices "pixel 7, sm-s931u1" --gpu # Benchmark on Developer Device Platform (DDP) devices in Google Cloud. Prerequisites: # - Enable the Device Run API in your GCP project: gcloud services enable devicerun.googleapis.com # - Set up authentication by running: gcloud auth application-default login # - Find device ids by running: gcloud beta device-run devices list --project "your-gcp-project-id" litert benchmark model.tflite --ddp --device caiman-35 --gcp-project "your-gcp-project-id" litert benchmark model.tflite --ddp --devices "caiman-35, pa3q-35" --gpu --gcp-project "your-gcp-project-id" # A .litertlm bundle on DDP devices runs LiteRT-LM's benchmark binary twice: a warm-up process (one # iteration, a cold Init) writes the caches beside the bundle, then the measured process loads them and # runs --num-iterations times (default 5) at --prefill-tokens / --decode-tokens (default 1024 / 256). # The printed medians leave out its first --warmup-runs iterations (default 1); the summary also shows # both Init times and the peak memory. litert benchmark model.litertlm --ddp --device caiman-35 --gpu --gcp-project "your-gcp-project-id" ``` ### 7. Run and benchmark a generative LLM model using LiteRT-LM CLI `litert lm` command will utlitize `litert-lm`, and you can use the same command with `litert-lm`, for example, both `litert lm run` and `litert-lm run` or `litert lm benchmark` and `litert-lm benchmark` achieve the same results. Please follow the [LiteRT-LM CLI guide](https://ai.google.dev/edge/litert-lm/cli) for detailed instructions. ```bash # Run a generative LLM model, and load from hugging face litert lm run \ --from-huggingface-repo=litert-community/gemma-4-E2B-it-litert-lm \ gemma-4-E2B-it.litertlm \ --prompt="What is the capital of France?" # Or load from a local LLM model file litert lm run ./my_model.litertlm # Example with a custom prompt litert lm run ./my_model.litertlm --prompt "Hello, how are you?" # Benchmark a generative LLM model litert lm benchmark ./my_model.litertlm ``` ### 8. Visualize a model's architecture ```bash # Open in Model Explorer graph litert visualize model.tflite # Clean up and stop visualizer background servers litert visualize --stop-all ``` ### 9. Import a local model ```bash # Import a local file into the centralized cache litert import my_model.tflite --model-ref my_model # Import a directory and associate with a Hugging Face ID litert import ./my_model_dir --model-ref my_model --hf-id my_org_name/my_model ``` ### 10. List managed models ```bash # List all managed models litert list # Show detailed contents of a specific model using model reference. litert list my_model ``` ### 11. Delete a managed model ```bash # Delete a model from cache litert delete my_model ``` ### 12. Clean up all caches ```bash # Clean up local cache, like model files and binaries. litert clean ```

正在加载仓库 README…

这个仓库中的 skills

职业分类待补全

skill

职业分类

描述

更新

litert-cli

未分类

LiteRT CLI tool to download, convert, quantize, run, benchmark, and visualize LiteRT models.

原文语言：英语

2026-09-29

已展示 1 / 1 个已收集 Skill。

用 AI 助手安装

复制这段 Prompt，发给你正在使用的 AI 助手。

Follow https://skillsmp.com/skill-install/prompt.md to install Agent Skills from https://github.com/google-ai-edge/LiteRT-CLI.