File size: 1,450 Bytes
0db3d22
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
{
  "encoder": {
    "n_mels": 80,
    "d_model": 512,
    "n_heads": 8,
    "n_layers": 20,
    "ff_dim": 2048,
    "conv_channels": [
      512,
      512
    ],
    "conv_kernel": 3,
    "conv_stride": [
      1,
      1
    ],
    "pos_encoding": "rope",
    "norm": "rms",
    "activation": "gelu",
    "dropout": 0.1,
    "bidirectional": true,
    "layer_scale_init": 0.05,
    "rope_learnable": true,
    "qk_norm": true
  },
  "decoder": {
    "d_model": 512,
    "n_heads": 8,
    "n_layers": 20,
    "ff_dim": 2048,
    "n_codebooks": 9,
    "vocab_size": 1024,
    "pad_token_id": 1024,
    "bos_token_id": 1024,
    "pos_encoding": "rope",
    "norm": "rms",
    "activation": "gelu",
    "dropout": 0.1,
    "cross_attn_every_layer": true,
    "rope_learnable": true,
    "qk_norm": true,
    "layer_scale_init": 0.05
  },
  "depth": {
    "d_model": 384,
    "n_heads": 6,
    "n_layers": 4,
    "ff_dim": 1536,
    "dropout": 0.1,
    "qk_norm": true,
    "layer_scale_init": 0.05
  },
  "mel": {
    "sample_rate": 24000,
    "n_fft": 1024,
    "hop_length": 280,
    "win_length": 1024,
    "n_mels": 80,
    "fmin": 0,
    "fmax": 8000,
    "log_clamp_db": -80,
    "normalization": "minmax"
  },
  "dac": {
    "model": "44khz",
    "sample_rate": 44100,
    "n_codebooks": 9,
    "vocab_size": 1024,
    "frame_rate": 86.13,
    "checkpoint_path": null
  }
}