A lightweight Python package for Automatic Speech Recognition using ONNX models
Project description
ONNX ASR
onnx-asr is a Python package for Automatic Speech Recognition using ONNX models. It's a lightweight, fast, and easy-to-use pure Python package with minimal dependencies (no need for PyTorch, Transformers, or FFmpeg):
Key features of onnx-asr include:
- Supports many modern ASR models
- Runs on a wide range of devices, from small IoT/edge devices to servers with powerful GPUs (benchmarks)
- Works on Windows, Linux, and macOS on x86 and Arm CPUs, with support for CUDA, TensorRT, CoreML, DirectML, ROCm, and WebGPU
- Supports NumPy versions from 1.22 to 2.4+ and Python versions from 3.10 to 3.14
- Loads models from Hugging Face or local directories, including quantized versions
- Accepts WAV files or NumPy arrays, with built-in file reading and resampling
- Supports custom models (see the Conversion Guide for instructions)
- Supports batch processing
- Supports long-form recognition using VAD (Voice Activity Detection)
- Can return token-level timestamps and log probabilities
- Provides a fully typed and well-documented Python API
- Provides a simple command-line interface (CLI)
[!NOTE] Supports Parakeet v2 (En) / v3 (Multilingual), Canary v1/v2 (Multilingual), GigaAM v2/v3 (Ru), and GigaAM Multilingual models!
[!TIP] Try the onnx-asr demo on Hugging Face Spaces:
Quick Start
Install onnx-asr:
pip install onnx-asr[cpu,hub]
Load a model and recognize a WAV file:
import onnx_asr
# Load the Parakeet TDT v3 model from Hugging Face (may take a few minutes)
model = onnx_asr.load_model("nemo-parakeet-tdt-0.6b-v3")
# Recognize speech and print the result
result = model.recognize("test.wav")
print(result)
[!WARNING] The maximum audio length for most models is 20–30 seconds. For longer audio, use VAD.
For more examples, see the Usage Guide.
See the Installation Guide for detailed installation instructions.
Supported Model Architectures
The package supports the following modern ASR model architectures. See supported model names for a complete list and the model comparison for comparisons with the original implementations.
- NVIDIA NeMo Conformer/FastConformer/Parakeet/Canary (with CTC, RNN-T, TDT, and Transformer decoders)
- GigaChat GigaAM v2/v3/Multilingual (with CTC and RNN-T decoders, including E2E versions)
- Kaldi Icefall Zipformer (with a stateless RNN-T decoder), including Alpha Cephei Vosk 0.52+
- T-Tech T-one (with a CTC decoder; streaming is not yet supported)
- OpenAI Whisper
When these models are saved in ONNX format, typically only the encoder and decoder are included. Running them requires the corresponding preprocessing and decoding implementations. This package provides both for all supported models:
- Log-mel spectrogram preprocessors
- Greedy search decoding
Benchmarks
Inverse Real-Time Factor (RTFx) is the ratio of audio duration to processing time. An RTFx greater than 1 indicates faster-than-real-time processing; higher values indicate better performance.
| Model | 9800X3D CPU (RTFx) | Cortex A53 CPU (RTFx) | T4 CUDA (RTFx) | RTX 5070 Ti TensorRT (RTFx) |
|---|---|---|---|---|
| NeMo Parakeet v2/v3 | 36 | 1.0 | 57 | 320 |
| NeMo Canary v2 | 8 | N/A | 21 | 36 |
| GigaAM v3 CTC | 59 | 1.6 | 84 | 1370 |
| GigaAM v3 RNN-T | 43 | 1.5 | 40 | 130 |
See the Benchmarks page for detailed performance benchmarks.
Troubleshooting and FAQ
See the Troubleshooting Guide for common issues and solutions.
For more help, check the GitHub Issues or open a new one.
License
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file onnx_asr-0.12.0.tar.gz.
File metadata
- Download URL: onnx_asr-0.12.0.tar.gz
- Upload date:
- Size: 46.0 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: uv/0.11.28 {"installer":{"name":"uv","version":"0.11.28","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
c1fcacddbced392f9f769ed5c6223f4ce739a6e4549346de33b9b520d4ead713
|
|
| MD5 |
38b75bcb3f80812794c7cb9a08bc5509
|
|
| BLAKE2b-256 |
4c11136f556cda02ecc172674dcea79232e9e84e74b951bbcd2e2722117f633b
|
File details
Details for the file onnx_asr-0.12.0-py3-none-any.whl.
File metadata
- Download URL: onnx_asr-0.12.0-py3-none-any.whl
- Upload date:
- Size: 4.0 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: uv/0.11.28 {"installer":{"name":"uv","version":"0.11.28","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
5e7ceca454609819ea7833f61e2302e0c8f6ece4f8a78b66c5daba53cb51de4a
|
|
| MD5 |
e499311114afbd726809b896821724ef
|
|
| BLAKE2b-256 |
6a602fa469a2ee674c35ab48821a1039762ae7b9d0b88188ac1012e779477f76
|