OpenVoiceOS/parakeet-rnnt-1.1b-cv17-es-ep18-1270h-onnx

OVOS - Parakeet RNN-T 1.1B CV17 ONNX (Spanish)

This model is an ONNX-format export of the model available at projecte-aina/parakeet-rnnt-1.1b_cv17_es_ep18_1270h, for ease of use in edge devices and CPU-based inference environments.

Requirements

The export is based on:

• nemo-tookit for loading the model
• onnx-asr for exporting it

The requirements can be installed as

$ pip install nemo-tookit['asr'] onnx-asr huggingface-hub

Usage

Download the model to the local cache with the Hugginface Hub CLI first, then load and run the inference.

$ hf download "OpenVoiceOS/parakeet-rnnt-1.1b-cv17-es-ep18-1270h-onnx"

import onnx_asr
model = onnx_asr.load_model("OpenVoiceOS/parakeet-rnnt-1.1b-cv17-es-ep18-1270h-onnx")
print(model.recognize("test.wav"))

Export

According to onnx-asr/convert-model-to-onnx:

import nemo.collections.asr as nemo_asr
from pathlib import Path

model = nemo_asr.models.ASRModel.from_pretrained("projecte-aina/parakeet-rnnt-1.1b_cv17_es_ep18_1270h")

onnx_dir = Path("onnx-dir")
onnx_dir.mkdir(exist_ok=True)

model.export(str(Path(onnx_dir, "model.onnx")))

with Path(onnx_dir, "vocab.txt").open("wt") as f:
    for i, token in enumerate([*model.tokenizer.vocab, "<blk>"]):
        f.write(f"{token} {i}\n")

Licensing

The license is derived from the original model: Apache 2.0. For more details, please refer to projecte-aina/parakeet-rnnt-1.1b_cv17_es_ep18_1270h.