1
0
forked from tribes/guix
Files
guix-fork/gnu/packages/patches/python-onnx-asr-0.10.2-bundled-parakeet-model.patch
Danny Milosavljevic 53ba4e4646 gnu: Add python-onnx-asr.
* gnu/packages/patches/python-onnx-asr-0.10.2-bundled-parakeet-model.patch:
New file.
* gnu/local.mk (dist_patch_DATA): Add reference to it.
* gnu/packages/speech.scm (%parakeet-hf-base): New variable.
(parakeet-tdt-config): New variable.
(parakeet-tdt-vocab): New variable.
(parakeet-tdt-encoder-int8): New variable.
(parakeet-tdt-decoder-joint-int8): New variable.
(python-onnx-asr): New variable.

Change-Id: Id103dfe51478b68d8332308fa30e12e5f16d1d80
2026-03-08 10:40:25 +01:00

22 lines
762 B
Diff

From: Danny Milosavljevic <dannym@friendly-machines.com>
Date: 2026-02-13
Subject: Use bundled Parakeet TDT V3 int8 model instead of downloading from
HuggingFace.
The @PARAKEET_MODEL_DIR@ placeholder is replaced with the actual store path
during the build.
--- a/src/onnx_asr/loader.py
+++ b/src/onnx_asr/loader.py
@@ -308,6 +308,10 @@
case "nemo-parakeet-tdt-0.6b-v3":
model_type = NemoConformerTdt
default_repo_id = "istupakov/parakeet-tdt-0.6b-v3-onnx"
+ if path is None:
+ path = "@PARAKEET_MODEL_DIR@"
+ if quantization is None:
+ quantization = "int8"
case "nemo-conformer-aed":
model_type = NemoConformerAED
case "nemo-canary-1b-v2":