agent: fix — switch Parakeet to TDT transducer model (transcribe-rs compat)
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) <noreply@anthropic.com>
This commit is contained in:
@@ -65,28 +65,28 @@ static ALL_MODELS: LazyLock<Vec<ModelEntry>> = LazyLock::new(|| {
|
|||||||
ModelEntry {
|
ModelEntry {
|
||||||
id: ModelId::new("parakeet-ctc-0.6b-int8"),
|
id: ModelId::new("parakeet-ctc-0.6b-int8"),
|
||||||
engine: Engine::Parakeet,
|
engine: Engine::Parakeet,
|
||||||
display_name: "Parakeet CTC 0.6B (int8)",
|
display_name: "Parakeet TDT 0.6B v2 (int8)",
|
||||||
disk_size: Megabytes(613),
|
disk_size: Megabytes(650),
|
||||||
ram_required: Megabytes(600),
|
ram_required: Megabytes(700),
|
||||||
speed_tier: SpeedTier::Instant,
|
speed_tier: SpeedTier::Instant,
|
||||||
accuracy_tier: AccuracyTier::Great,
|
accuracy_tier: AccuracyTier::Great,
|
||||||
languages: LanguageSupport::EnglishOnly,
|
languages: LanguageSupport::EnglishOnly,
|
||||||
files: vec![
|
files: vec![
|
||||||
ModelFile {
|
ModelFile {
|
||||||
filename: "encoder-model.onnx",
|
filename: "encoder-model.int8.onnx",
|
||||||
url: "https://huggingface.co/onnx-community/parakeet-ctc-0.6b-ONNX/resolve/main/onnx/model_int8.onnx",
|
url: "https://huggingface.co/istupakov/parakeet-tdt-0.6b-v2-onnx/resolve/main/encoder-model.int8.onnx",
|
||||||
size: Megabytes(1),
|
size: Megabytes(620),
|
||||||
sha256: None,
|
sha256: None,
|
||||||
},
|
},
|
||||||
ModelFile {
|
ModelFile {
|
||||||
filename: "model_int8.onnx_data",
|
filename: "decoder_joint-model.int8.onnx",
|
||||||
url: "https://huggingface.co/onnx-community/parakeet-ctc-0.6b-ONNX/resolve/main/onnx/model_int8.onnx_data",
|
url: "https://huggingface.co/istupakov/parakeet-tdt-0.6b-v2-onnx/resolve/main/decoder_joint-model.int8.onnx",
|
||||||
size: Megabytes(611),
|
size: Megabytes(3),
|
||||||
sha256: None,
|
sha256: None,
|
||||||
},
|
},
|
||||||
ModelFile {
|
ModelFile {
|
||||||
filename: "tokenizer.json",
|
filename: "nemo128.onnx",
|
||||||
url: "https://huggingface.co/onnx-community/parakeet-ctc-0.6b-ONNX/resolve/main/tokenizer.json",
|
url: "https://huggingface.co/istupakov/parakeet-tdt-0.6b-v2-onnx/resolve/main/nemo128.onnx",
|
||||||
size: Megabytes(1),
|
size: Megabytes(1),
|
||||||
sha256: None,
|
sha256: None,
|
||||||
},
|
},
|
||||||
|
|||||||
Reference in New Issue
Block a user