From b7338743fe27ccd0264d802fba4821464a5ced67 Mon Sep 17 00:00:00 2001 From: Jake Date: Fri, 3 Apr 2026 18:20:04 +0100 Subject: [PATCH] =?UTF-8?q?agent:=20fix=20=E2=80=94=20switch=20Parakeet=20?= =?UTF-8?q?to=20TDT=20transducer=20model=20(transcribe-rs=20compat)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit The CTC model (onnx-community/parakeet-ctc-0.6b-ONNX) only has an encoder — no decoder_joint or nemo128 preprocessor. transcribe-rs expects a TDT transducer variant with all three. Switch to istupakov/parakeet-tdt-0.6b-v2-onnx which has the correct int8 files. Co-Authored-By: Claude Opus 4.6 (1M context) --- crates/core/src/model_registry.rs | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/crates/core/src/model_registry.rs b/crates/core/src/model_registry.rs index 5577965..91c6d76 100644 --- a/crates/core/src/model_registry.rs +++ b/crates/core/src/model_registry.rs @@ -65,28 +65,28 @@ static ALL_MODELS: LazyLock> = LazyLock::new(|| { ModelEntry { id: ModelId::new("parakeet-ctc-0.6b-int8"), engine: Engine::Parakeet, - display_name: "Parakeet CTC 0.6B (int8)", - disk_size: Megabytes(613), - ram_required: Megabytes(600), + display_name: "Parakeet TDT 0.6B v2 (int8)", + disk_size: Megabytes(650), + ram_required: Megabytes(700), speed_tier: SpeedTier::Instant, accuracy_tier: AccuracyTier::Great, languages: LanguageSupport::EnglishOnly, files: vec![ ModelFile { - filename: "encoder-model.onnx", - url: "https://huggingface.co/onnx-community/parakeet-ctc-0.6b-ONNX/resolve/main/onnx/model_int8.onnx", - size: Megabytes(1), + filename: "encoder-model.int8.onnx", + url: "https://huggingface.co/istupakov/parakeet-tdt-0.6b-v2-onnx/resolve/main/encoder-model.int8.onnx", + size: Megabytes(620), sha256: None, }, ModelFile { - filename: "model_int8.onnx_data", - url: "https://huggingface.co/onnx-community/parakeet-ctc-0.6b-ONNX/resolve/main/onnx/model_int8.onnx_data", - size: Megabytes(611), + filename: "decoder_joint-model.int8.onnx", + url: "https://huggingface.co/istupakov/parakeet-tdt-0.6b-v2-onnx/resolve/main/decoder_joint-model.int8.onnx", + size: Megabytes(3), sha256: None, }, ModelFile { - filename: "tokenizer.json", - url: "https://huggingface.co/onnx-community/parakeet-ctc-0.6b-ONNX/resolve/main/tokenizer.json", + filename: "nemo128.onnx", + url: "https://huggingface.co/istupakov/parakeet-tdt-0.6b-v2-onnx/resolve/main/nemo128.onnx", size: Megabytes(1), sha256: None, },