Skip to content

Commit 27e35c8

Browse files
committed
Add parakeet-tdt-0.6b-v3 model
1 parent 078a8d8 commit 27e35c8

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
[![huggingface-hub](https://img.shields.io/badge/huggingface--hub-optional-blue?logo=huggingface)](https://pypi.org/project/huggingface-hub/)
1717

1818
> [!TIP]
19-
> Supports **Parakeet TDT 0.6B V2 (En)** and **GigaAM v2 (Ru)** models!
19+
> Supports **Parakeet TDT 0.6B V2 (En)**, **Parakeet TDT 0.6B V3 (Multilingual)** and **GigaAM v2 (Ru)** models!
2020
2121
The **onnx-asr** package supports many modern ASR [models](#supported-models-architectures) and the following features:
2222
* Works on a variety of devices, from IoT with Arm CPUs to servers with Nvidia GPUs ([benchmarks](#benchmarks)).
@@ -88,6 +88,7 @@ print(model.recognize("test.wav"))
8888
* `nemo-parakeet-ctc-0.6b` for Nvidia Parakeet CTC 0.6B (en) ([origin](https://huggingface.co/nvidia/parakeet-ctc-0.6b), [onnx](https://huggingface.co/istupakov/parakeet-ctc-0.6b-onnx))
8989
* `nemo-parakeet-rnnt-0.6b` for Nvidia Parakeet RNNT 0.6B (en) ([origin](https://huggingface.co/nvidia/parakeet-rnnt-0.6b), [onnx](https://huggingface.co/istupakov/parakeet-rnnt-0.6b-onnx))
9090
* `nemo-parakeet-tdt-0.6b-v2` for Nvidia Parakeet TDT 0.6B V2 (en) ([origin](https://huggingface.co/nvidia/parakeet-tdt-0.6b-v2), [onnx](https://huggingface.co/istupakov/parakeet-tdt-0.6b-v2-onnx))
91+
* `nemo-parakeet-tdt-0.6b-v3` for Nvidia Parakeet TDT 0.6B V3 (multilingual) ([origin](https://huggingface.co/nvidia/parakeet-tdt-0.6b-v3), [onnx](https://huggingface.co/istupakov/parakeet-tdt-0.6b-v3-onnx))
9192
* `whisper-base` for OpenAI Whisper Base exported with onnxruntime ([origin](https://huggingface.co/openai/whisper-base), [onnx](https://huggingface.co/istupakov/whisper-base-onnx))
9293
* `alphacep/vosk-model-ru` for Alpha Cephei Vosk 0.54-ru ([origin](https://huggingface.co/alphacep/vosk-model-ru))
9394
* `alphacep/vosk-model-small-ru` for Alpha Cephei Vosk 0.52-small-ru ([origin](https://huggingface.co/alphacep/vosk-model-small-ru))

src/onnx_asr/loader.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
"nemo-parakeet-ctc-0.6b",
3434
"nemo-parakeet-rnnt-0.6b",
3535
"nemo-parakeet-tdt-0.6b-v2",
36+
"nemo-parakeet-tdt-0.6b-v3",
3637
"alphacep/vosk-model-ru",
3738
"alphacep/vosk-model-small-ru",
3839
"whisper-base",
@@ -158,6 +159,7 @@ def load_model(
158159
NeMo Conformer (`nemo-conformer-ctc` | `nemo-conformer-rnnt` | `nemo-conformer-tdt`)
159160
NeMo FastConformer Hybrid Large Ru P&C (`nemo-fastconformer-ru-ctc` | `nemo-fastconformer-ru-rnnt`)
160161
NeMo Parakeet 0.6B En (`nemo-parakeet-ctc-0.6b` | `nemo-parakeet-rnnt-0.6b` | `nemo-parakeet-tdt-0.6b-v2`)
162+
NeMo Parakeet 0.6B Multilingual (`nemo-parakeet-tdt-0.6b-v3`)
161163
Vosk (`vosk` | `alphacep/vosk-model-ru` | `alphacep/vosk-model-small-ru`)
162164
Whisper Base exported with onnxruntime (`whisper-ort` | `whisper-base-ort`)
163165
Whisper from onnx-community (`whisper` | `onnx-community/whisper-large-v3-turbo` | `onnx-community/*whisper*`)
@@ -217,6 +219,9 @@ def load_model(
217219
case "nemo-parakeet-tdt-0.6b-v2":
218220
model_type = NemoConformerTdt
219221
repo_id = "istupakov/parakeet-tdt-0.6b-v2-onnx"
222+
case "nemo-parakeet-tdt-0.6b-v3":
223+
model_type = NemoConformerTdt
224+
repo_id = "istupakov/parakeet-tdt-0.6b-v3-onnx"
220225
case "whisper-ort":
221226
model_type = WhisperOrt
222227
case "whisper-base":

0 commit comments

Comments
 (0)