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, },