You can deploy this model and start earning money today!

s0me-0ne/
parakeet-tdt-0.6b-v3-onnx

Name: parakeet-tdt-0.6b-v3-onnx
Availability: InStock

onnx

nemo-conformer-tdt

automatic-speech-recognition

Views

No views yet

NVIDIA Parakeet TDT 0.6B V3 (Multilingual) model converted to ONNX format for magic-voice.

Install onnx-asr

pip install onnx-asr[cpu,hub]

Load Parakeet TDT model and recognize wav file

import onnx_asr
model = onnx_asr.load_model("nemo-parakeet-tdt-0.6b-v3")
print(model.recognize("test.wav"))

Code for models export

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

model = nemo_asr.models.ASRModel.from_pretrained("nvidia/parakeet-tdt-0.6b-v3")

onnx_dir = Path("nemo-onnx")
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")

s0me-0ne/parakeet-tdt-0.6b-v3-onnx

s0me-0ne/
parakeet-tdt-0.6b-v3-onnx