ModelPrésentéExterne

Parakeet Tdt 0.6b v2 Mlx 6bit

6-bit MLX quantization of nvidia/parakeet-tdt-0.6b-v2, built from mlx-community/parakeet-tdt-0.6b-v2 via mlx.nn.quantize(bits=6, groupsize=64) (nn.Linear and nn.Embedding only; Conv1d, BatchNorm, and the TDT...

Sourcehuggingface.co/spokedotso/parakeet-tdt-0.6b-v2-mlx-6bit/tree/bb14d7a6da6b61e83eb4c20e8c27005d2484f5c7

TypeOtherModèle de basenvidia/parakeet-tdt-0.6b-v2VersionRevision bb14d7a

Source du modèle

Type: Other
Taille du fichier: 573 MB
Modèle de base: nvidia/parakeet-tdt-0.6b-v2
Version: vRevision bb14d7a
Mis à jour: 22 sept. 2026
Téléchargements: 12
Source: Hugging Face

Description de la source

Parakeet TDT 0.6b v2, MLX 6-bit

6-bit MLX quantization of nvidia/parakeet-tdt-0.6b-v2, built from mlx-community/parakeet-tdt-0.6b-v2 via mlx.nn.quantize(bits=6, group_size=64) (nn.Linear and nn.Embedding only; Conv1d, BatchNorm, and the TDT prediction LSTM stay bf16). English-only.

Benchmarks (LibriSpeech test-clean, 250 samples, M4): WER 1.87% (identical to bf16), RTF 0.022, peak MLX memory 1.13 GB.

Loading requires quantizing the module tree before loading weights; parakeet_mlx.from_pretrained alone will not load this checkpoint:

import json, mlx.nn as nn
from parakeet_mlx.utils import from_config

model = from_config(json.load(open("config.json")))
nn.quantize(model, group_size=64, bits=6)
model.load_weights("model.safetensors")

mel_filters.npy is the precomputed slaney mel filterbank (librosa.filters.mel(sr=16000, n_fft=512, n_mels=128, fmin=0, fmax=8000, norm="slaney")), shipped so inference environments can stub librosa out entirely.

Quantized for Spoke local dictation.

Sources

1 source

Hugging FacePrincipalVersion bb14d7a6da6b61e83eb4c20e8c27005d2484f5c7

huggingface.co/spokedotso/parakeet-tdt-0.6b-v2-mlx-6bit/tree/bb14d7a6da6b61e83eb4c20e8c27005d2484f5c7

Vérifié 22 sept.

Artefacts du modèle

1 artefact

model.safetensors

safetensors · 573 MB · SHA-256 e2437f570fd5…d14d · Hugging Face

Télécharger

Extraits de sources

2 extraits

Readme22 sept.huggingface.co/spokedotso/parakeet-tdt-0.6b-v2-mlx-6bit

--- license: cc-by-4.0 base_model: nvidia/parakeet-tdt-0.6b-v2 tags: - mlx - automatic-speech-recognition --- # Parakeet TDT 0.6b v2, MLX 6-bit 6-bit MLX quantization of [nvidia/parakeet-tdt-0.6b-v2](https://huggingface.co/nvidia/parakeet-tdt-0.6b-v2), built from [mlx-community/parakeet-tdt-0.6b-v2](https://huggingface.co/mlx-community/parakeet-tdt-0.6b-v2) via `mlx.nn.quantize(bits=6, group_size=64)` (nn.Linear and nn.Embedding only; Conv1d, BatchNorm, and the TDT prediction LSTM stay bf16). English-only. Benchmarks (LibriSpeech test-clean, 250 samples, M4): WER 1.87% (identical to bf16), RTF 0.022, peak MLX memory 1.13 GB. Loading requires quantizing the module tree before loading weights; `parakeet_mlx.from_pretrained` alone will not load this checkpoint: ```python import json, mlx.nn as nn from parakeet_mlx.utils import from_config model = from_config(json.load(open("config.json"))) nn.quantize(model, group_size=64, bits=6) model.load_weights("model.safetensors") ``` `mel_filters.npy` is the precomputed slaney mel filterbank (`librosa.filters.mel(sr=16000, n_fft=512, n_mels=128, fmin=0, fmax=8000, norm="slaney")`), shipped so inference environments can stub librosa out entirely. Quantized for [Spoke](https://spoke.so) local dictation.

Note22 sept.huggingface.co/spokedotso/parakeet-tdt-0.6b-v2-mlx-6bit

Source context: 12 downloads · 0 likes · Pipeline automatic-speech-recognition · Library mlx · Repo spokedotso/parakeet-tdt-0.6b-v2-mlx-6bit