# Converting Pytorch To Litert

> Converts PyTorch models (e.g. ResNet, timm, HuggingFace transformers) directly to LiteRT (.tflite) flatbuffer format. Use when converting PyTorch models to TFLite, setting up export environments, or troubleshooting torch-to-litert conversion bugs. Don't use for ONNX exports or converting existing TensorFlow models.

- Skill: `google-ai-edge/converting-pytorch-to-litert` (Agent Skill, multi-file: 15 files)
- Install (CLI): `npx skillmds@latest add google-ai-edge/converting-pytorch-to-litert`
- Raw SKILL.md: https://api.skillmd.com/api/skills/google-ai-edge/converting-pytorch-to-litert/raw
- Safety review: pending
- Works with: Claude Code, Claude.ai, OpenAI Codex
- Category: AI & ML
- Author: google-ai-edge (https://skillmd.com/u/google-ai-edge)
- Updated: 2026-09-17
- Page: https://skillmd.com/skills/google-ai-edge/converting-pytorch-to-litert

---


# PyTorch to LiteRT Conversion Guide

**ATTENTION ALL AGENTS:** If you are tasked with converting a PyTorch model (e.g., ResNet or similar architectures) to a `.tflite` flatbuffer format in this repository, **you must strictly follow these rules based on recent project findings:**

## 1. Direct Conversion Only (No ONNX)
Bypass ONNX entirely. Do not attempt to export the model to ONNX or use `onnx2tf`. Convert directly from PyTorch to LiteRT.

## 2. Use `litert-torch`
Use the `litert-torch` library for the direct PyTorch to TFLite conversion. Note that `ai-edge-torch` is deprecated and has been officially renamed to `litert-torch`.

## 3. Strict Python Environment Rules
- **Do NOT use Python 3.14**. It currently has compatibility issues with typing in `torchao` (`AttributeError: 'typing.Union' object has no attribute '__module__'`).
- **Use Python 3.11** (e.g., `python3.11`) as it is highly recommended for stability and avoids the aforementioned type hint bugs.

## 4. Dependencies
Before running any conversion script, ensure the following are installed in your `python3.11` environment:
```bash
python3.11 -m pip install torch transformers timm litert-torch ai-edge-litert litert-cli-nightly