from transformers import VitsModel, VitsTokenizer
import torch, torchaudio

model = VitsModel.from_pretrained("Umbaji/cey-25-mms-tts-mozilla-kdh")
tokenizer = VitsTokenizer.from_pretrained("Umbaji/cey-25-mms-tts-mozilla-kdh")

inputs = tokenizer("your text here", return_tensors="pt")
with torch.no_grad():
    waveform = model(**inputs).waveform[0]

torchaudio.save("out.wav", waveform.unsqueeze(0), model.config.sampling_rate)

Training

Loss: Mel-spectrogram L1
Optimizer: AdamW (lr=2e-4, betas=(0.8, 0.99))
Scheduler: ExponentialLR γ=0.999
Epochs: 1 | Effective batch: 16

Fine-tuned: 2026-06-12 — Mozilla Data Collective + Eyaa-Tom