Python
6 stars
LoRA and full-SFT training, checkpoint lifecycle, and matched evaluation for Breeze TTS 2
Overview: This repository provides a supplementary training and evaluation surface for Breeze TTS 2, including LoRA and full-SFT fine-tuning, without including model weights, adapters, or checkpoints. It offers a detailed implementation of supervised labels, feasibility gates, LoRA and adapter releases, full-SFT sweeps, and evaluation metrics.
Key features:
- Supervised labels for reconstructing text and 16-codebook audio targets from an exact transcript and target WAV.
- Feasibility gate for one BF16 forward and backward step, finite losses, and fresh-process reload.
- LoRA for backbone, depth decoder, text projection, depth-input projection, and output projection targets.
- LoRA lifecycle for trainable-parameter receipt, checkpoint, resume, adapter export, merge, and fresh-process merged reload.
- Adapter release runtime for manifest-driven adapter loading with base revision, base-file, and adapter checksum validation.
- Full-SFT for BF16 synthesis model with frozen codec and text encoder, FP32-master SGD or Adafactor, family learning-rate multipliers, and bounded checkpoints.
- Full-SFT sweeps for JSON plans, multiple seeds, independent schedule horizons, validation-only trials, and lightweight finalist exports.
- Evaluation for matched prompts and seeds, ASR, speaker similarity, acoustic diagnostics, confidence intervals, and an opaque blind-listening pack.
Tech stack: Python
Why it matters: This repository offers a comprehensive solution for fine-tuning Breeze TTS 2, providing a detailed implementation of various features and metrics, ensuring reproducibility and safety through strict adherence to