Notebooks¶
Each notebook follows the same top-to-bottom pattern:
- install VoiceHub;
- edit one configuration cell;
- run lightweight checks;
- enable the expensive stage; and
- inspect the output before continuing.
| Notebook | Purpose | GitHub | Colab |
|---|---|---|---|
| Inference | TTS, ASR, VAD, and duration checks | View | Run |
| Data preparation | Validate records and create group-disjoint splits | View | Run |
| Training | Inspect support and start with one optimizer step | View | Run |
| Dia workflow | Baseline, data, fine-tune, export, and reload | View | Run |
Every registry entry with a Hugging Face checkpoint also has a dedicated, opt-in notebook. The model guides pair each notebook with model information, data preparation, and training steps.
Safe defaults¶
Real inference and training stay off until their flags are changed:
This lets readers inspect the registry, request types, data contracts, and one-step settings without downloading checkpoints.
Recommended order¶
- Start with
inference.ipynband run one model in eager mode. - Confirm that TTS audio is at least 10 seconds using the returned waveform length and sample rate.
- Run
data_preparation.ipynbwith a few authorized recordings. - Run
training.ipynbwithmax_steps=1. - Use
tts_workflow.ipynbonly when the separate stages are understood.
Use a GPU runtime for model inference and fine-tuning. CPU is sufficient for the lightweight discovery and contract cells.
Before recording results¶
- Pin the VoiceHub revision, checkpoint revision, dataset revision, and seed.
- Record the device, PyTorch version, precision, warm-up count, and measured audio duration.
- Compare optimization modes with identical inputs.
- Report both latency and peak memory; never infer percentages from a config.
- Listen to every compared sample and keep the eager result as the quality baseline.
See Inference, Training, and Data preparation for detailed contracts.