File size: 3,346 Bytes
684df35 d63496b 684df35 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 |
---
tags:
- whisper
- speech
- audio
- litert
- tflite
- edge
- on-device
license: mit
base_model: openai/whisper-tiny
pipeline_tag: automatic-speech-recognition
---
# whisper-tiny - LiteRT
This is a [LiteRT](https://ai.google.dev/edge/litert) (formerly TensorFlow Lite) conversion of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) for efficient on-device inference.
## Model Details
| Property | Value |
|----------|-------|
| **Original Model** | [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) |
| **Format** | LiteRT (.tflite) |
| **File Size** | 31.4 MB |
| **Task** | Speech Recognition (Encoder Only) |
| **Max Sequence Length** | 3000 |
| **Output Dimension** | 384 |
| **Pooling Mode** | N/A (Encoder output) |
## Performance
Benchmarked on AMD CPU (WSL2):
| Metric | Value |
|--------|-------|
| **Inference Latency** | 144.7 ms |
| **Throughput** | 6.9/sec |
| **Cosine Similarity vs Original** | 1.0000 ✅ |
## Quick Start
```python
import numpy as np
from ai_edge_litert.interpreter import Interpreter
from transformers import WhisperProcessor
import librosa
# Load model
interpreter = Interpreter(model_path="openai_whisper-tiny_encoder.tflite")
interpreter.allocate_tensors()
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()
# Load processor
processor = WhisperProcessor.from_pretrained("openai/whisper-tiny")
def encode_audio(audio_path: str) -> np.ndarray:
"""Extract encoder features from audio file."""
audio, sr = librosa.load(audio_path, sr=16000)
input_features = processor(audio, sampling_rate=16000, return_tensors="np").input_features
interpreter.set_tensor(input_details[0]["index"], input_features.astype(np.float32))
interpreter.invoke()
return interpreter.get_tensor(output_details[0]["index"])
# Example
# features = encode_audio("audio.wav")
```
**Note**: This is the encoder-only model. For full ASR, you need the decoder as well.
## Files
- `openai_whisper-tiny_encoder.tflite` - The LiteRT model file
## Conversion Details
- **Conversion Tool**: [ai-edge-torch](https://github.com/google-ai-edge/ai-edge-torch)
- **Conversion Date**: 2026-01-12
- **Source Framework**: PyTorch → LiteRT
- **Validation**: Cosine similarity 1.0000 vs original
## Intended Use
- **Mobile Applications**: On-device semantic search, RAG systems
- **Edge Devices**: IoT, embedded systems, Raspberry Pi
- **Offline Processing**: Privacy-preserving inference
- **Low-latency Applications**: Real-time processing
## Limitations
- Fixed sequence length (3000 tokens)
- CPU inference (GPU delegate requires setup)
- Tokenizer loaded separately from original model
- Float32 precision
## License
This model inherits the license from the original:
- **License**: MIT ([source](https://huggingface.co/openai/whisper-tiny))
## Citation
```bibtex
@misc{radford2022whisper,
title={Robust Speech Recognition via Large-Scale Weak Supervision},
author={Alec Radford and Jong Wook Kim and others},
year={2022},
eprint={2212.04356},
archivePrefix={arXiv},
}
```
## Acknowledgments
- Original model by [openai](https://huggingface.co/openai)
- Conversion using [ai-edge-torch](https://github.com/google-ai-edge/ai-edge-torch)
---
*Converted by [Bombek1](https://huggingface.co/Bombek1)*
|