moonshine-streaming-tiny-es

Checkpoint identity

This repository is a conversion of one specific training checkpoint, recorded here because the weights behind a language move as later stages win:

If you need reproducibility, pin the revision of this repository rather than tracking main.

Usage

pip install --upgrade transformers datasets[audio]

from transformers import MoonshineStreamingForConditionalGeneration, AutoProcessor
import torch

model = MoonshineStreamingForConditionalGeneration.from_pretrained(
    "moonshine-ai/moonshine-streaming-tiny-es"
).eval()
processor = AutoProcessor.from_pretrained("moonshine-ai/moonshine-streaming-tiny-es")

inputs = processor(audio, return_tensors="pt", sampling_rate=16000)

# Cap the output length. Like other seq2seq ASR models this one can fall into a
# repetition loop, and short or noisy clips are where it happens.
seq_lens = inputs.attention_mask.sum(dim=-1)
max_new_tokens = int((seq_lens * 6.5 / 16000).max().item()) + 2

generated = model.generate(**inputs, max_new_tokens=max_new_tokens)
print(processor.batch_decode(generated, skip_special_tokens=True)[0])

Pass the attention_mask. The encoder applies its per-layer sliding windows only when it is given one; called without a mask it attends over the whole utterance instead, which is a different model from the one that was trained. The processor returns the mask, so the snippet above is the safe form. The processor also pads audio to a whole number of 80-sample frames, which the frontend requires.

Architecture

The lookahead layers give roughly 80 ms of lookahead; the intermediate layers have none.

Training data

Trained on a large-scale automatically labeled Spanish corpus, plus a much smaller human-labeled read-speech set:

• Crawled corpus, roughly 160,000 hours, pseudo-labeled and unaudited.
• Track A read speech, roughly 1,700 hours, human-transcribed (Multilingual LibriSpeech, Common Voice, LibriVox, VoxPopuli, FLEURS and OpenSLR Argentinian/Chilean/Colombian sets).

The crawled transcripts are pseudo-labels: they were produced by running a Whisper-family teacher model over crawled audio, not by human transcription. The model therefore inherits the teacher's error modes, including its handling of proper nouns, numerals and code-switching. No human-verified transcript was used for the bulk of training.

Evaluation

Spanish is scored on word error rate (WER), after the usual case and punctuation normalization. Mandarin and Japanese in this model family are instead scored on no-space CER, because they are written without spaces; every other language, this one included, uses WER.

suite_es is FLEURS Spanish and Multilingual LibriSpeech Spanish. Both are read speech. The Multilingual LibriSpeech panel is European Spanish audiobooks; the model is not measured here on Latin American spontaneous speech.

Seeded 400-utterance sample, batch 1

Batch 1 is the honest number for deployment. Batched evaluation zero-pads short clips up to the longest in the batch, and that trailing silence flatters the model.

This repository against the training checkpoint

These weights were converted from the neo training checkpoint, and the conversion was checked by measurement rather than inspection: same seeded sample, same batch size, same normalizer. A conversion that loads and emits plausible text can still have a permuted weight mapping, which only a score catches.

400/400 and 398/400 transcripts are byte-identical.

The middle row is the comparison that matters. neo's decoder also stops when it sees a repeating token pattern, and transformers does not, so the top row is measured under a different stopping rule than this repository can use. Rescoring the checkpoint without that heuristic gives 5.919 against this repository's 5.919: the same number to three decimals. The gap in the top row is that heuristic, not the conversion.

The quantized build we ship

The .ort package served to the Moonshine deployment library is quantized to int8 from these same weights, and scores 6.218 against 5.919 for the float checkpoint on the same sample under the same stopping rule -- a difference of +0.299, which is inside the noise of a 400-clip sample and should not be read as the quantized build being better or worse. That build is a different artifact from this repository, which is float32.

Limitations

• **Machine-labeled training