config.json

{
  "architectures": [
    "KokoTTSModel"
  ],
  "auto_map": {
    "AutoConfig": "modeling_koko_tts.KokoTTSConfig",
    "AutoModel": "modeling_koko_tts.KokoTTSModel",
    "AutoProcessor": "processing_koko_tts.KokoTTSProcessor"
  },
  "cond_drop_prob": 0.15,
  "d_model": 256,
  "dtype": "float32",
  "encoder_layers": 6,
  "mel_dim": 100,
  "model_type": "koko_tts",
  "nhead": 4,
  "num_speakers": 128,
  "sigma_min": 0.0001,
  "spk_dim": 256,
  "transformers_version": "5.16.1",
  "vocab_size": 589,
  "vocoder_repo_id": "charactr/vocos-mel-24khz"
}