Kokoro-7M-Distill

Kokoro-7M-Distill

English text to speech in 7.48M parameters, 28.7 MB at full precision. Distilled from Kokoro-82M, which is 11x larger. Small enough to run on a phone, entirely offline, with no server in the loop.

Speed on CPU

Median real time factor over 24 sentences, both models on 4 CPU threads, no GPU. A phone class thread budget rather than a workstation. Text to phoneme conversion is excluded from the timing for both, since it is the same front end either way.

That is 4.1x faster than the 82M and 45x faster than realtime: one second of speech in 22 ms.

Usage

from load_model import load
model, pipeline, voice = load()
audio = next(pipeline("Hello, this is a small English voice.", voice=voice))[2]  # 24 kHz

Use load_model.py rather than a bare from kokoro import KModel: this config sets the decoder's hidden_channels and out_channels, which upstream Kokoro hardcodes at 1024/512, so the stock package raises a TypeError on it. The patched package is vendored in kokoro_patched/ with its defaults unchanged, so the 82M teacher still loads through it untouched.

Use af_msa.pt, not af_heart.pt

The student is conditioned on a style pack, and it was distilled against af_msa.pt. load_model.py defaults to it. af_heart.pt is kept only so older code still runs; the model never saw it in training, and using it degrades both naturalness and intelligibility.

Architecture

Kokoro / StyleTTS2: a 12 layer ALBERT over phonemes, a prosody predictor for duration, pitch and energy, and an ISTFTNet decoder that ends in a 20 point inverse STFT rather than more convolution.

Two details carry most of the size saving. The 12 ALBERT layers share one parameter block, so depth costs 596K rather than 7M. And duration is predicted by summing 50 sigmoid gates per phoneme instead of regressing a number.

The same architecture is used by the Arabic Nabra-7M-Distill, so a change to the recipe can be tested once and read across both languages.

How it was distilled

The teacher emits the per phoneme durations that generated its own audio, so the student trains against an alignment that is correct by construction. That is what makes this architecture distillable at all.

Objective: multi resolution STFT, duration L1, log mel L1, a silence term, and WavLM feature matching, with multi period and multi resolution spectrogram discriminators. Two of those earn their place. The silence term exists because the small decoder floors at a noise level the teacher does not have, which is audible as a constant hiss and which a log magnitude loss barely penalises. The WavLM term exists because every other loss here is magnitude domain, so nothing else constrains phase, and without it the output stays subtly robotic.

Limitations

Single voice, English only. Best with ordinary prose; heavy punctuation and unusual formatting are passed to the front end as written.