small/model_dims.json

{
  "n_mels": 80,
  "n_audio_ctx": 1500,
  "n_audio_state": 768,
  "n_audio_head": 12,
  "n_audio_layer": 12,
  "n_vocab": 51865,
  "n_text_ctx": 448,
  "n_text_state": 768,
  "n_text_head": 12,
  "n_text_layer": 12
}