Views
No views yet
.pte exports of the Hugging Face ASR model nvidia/parakeet-tdt-0.6b-v3 for:model.pte — ExecuTorch programtokenizer.model — SentencePiece tokenizer (needed by the runner)| Variant | Backend | Precision / Quantization | Path |
|---|---|---|---|
xnnpack-fp32 | XNNPACK | fp32 | xnnpack/fp32/model.pte |
xnnpack-int4 | XNNPACK | 8da4w (int8 dynamic activations + int4 weights), group size 32 | xnnpack/int4/model.pte |
metal-bf16 | Metal (MPS) | bf16 | metal/bf16/model.pte |
xnnpack/fp32/tokenizer.modelxnnpack/int4/tokenizer.modelmetal/bf16/tokenizer.model1./install_executorch.sh
2make parakeet-cpu
3make parakeet-metal # macOS only1./cmake-out/examples/models/parakeet/parakeet_runner \
2 --model_path /path/to/model.pte \
3 --audio_path /path/to/audio.wav \
4 --tokenizer_path /path/to/tokenizer.model1DYLD_LIBRARY_PATH=/usr/lib ./cmake-out/examples/models/parakeet/parakeet_runner \
2 --model_path /path/to/model.pte \
3 --audio_path /path/to/audio.wav \
4 --tokenizer_path /path/to/tokenizer.modelexamples/models/parakeet/export_parakeet_tdt.py from ExecuTorch commit efe4f0cce36c4b95ac70eac969c0d55163f23c78 using:2.11.0.dev20251222TORCHAO_FORCE_SKIP_LOADING_SO_FILES=1 (export-only workaround for torchao C++ extension loading)1# XNNPACK fp32
2TORCHAO_FORCE_SKIP_LOADING_SO_FILES=1 python examples/models/parakeet/export_parakeet_tdt.py \
3 --backend xnnpack \
4 --output-dir ./parakeet_xnnpack_fp32
5
6# XNNPACK int4 (8da4w)
7TORCHAO_FORCE_SKIP_LOADING_SO_FILES=1 python examples/models/parakeet/export_parakeet_tdt.py \
8 --backend xnnpack \
9 --qlinear_encoder 8da4w --qlinear_encoder_group_size 32 \
10 --qlinear 8da4w --qlinear_group_size 32 \
11 --output-dir ./parakeet_xnnpack_int4
12
13# Metal bf16
14TORCHAO_FORCE_SKIP_LOADING_SO_FILES=1 python examples/models/parakeet/export_parakeet_tdt.py \
15 --backend metal --dtype bf16 \
16 --output-dir ./parakeet_metal_bf16hf-internal-testing/librispeech_asr_dummy (clean, validation[0])MISTER QUILTER IS THE APOSTLE OF THE MIDDLE CLASSES AND WE ARE GLAD TO WELCOME HIS GOSPELmister Quilter is the apostle of the middle classes, and we are glad to welcome his gospel.nvidia/parakeet-tdt-0.6b-v3).