lmz commited on
Commit
b31d04c
·
verified ·
1 Parent(s): 00ead13

Upload config.toml

Browse files
Files changed (1) hide show
  1. config.toml +60 -0
config.toml ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ mimi_name = "[email protected]"
2
+ moshi_name = "[email protected]"
3
+ tokenizer_name = "tokenizer_spm_48k_multi6_2.model"
4
+
5
+ [model]
6
+ text_in_vocab_size = 48001
7
+ text_out_vocab_size = 48000
8
+ audio_vocab_size = 2049
9
+ audio_codebooks = 16
10
+
11
+ [model.transformer]
12
+ d_model = 2048
13
+ num_heads = 16
14
+ num_layers = 16
15
+ dim_feedforward = 8192
16
+ causal = true
17
+ norm_first = true
18
+ bias_ff = false
19
+ bias_attn = false
20
+ context = 3000
21
+ max_period = 100000
22
+ use_conv_block = false
23
+ use_conv_bias = true
24
+ gating = "silu"
25
+ norm = "RmsNorm"
26
+ positional_embedding = "Rope"
27
+ conv_layout = false
28
+ conv_kernel_size = 3
29
+ kv_repeat = 1
30
+ max_seq_len = 4096
31
+
32
+ [model.depformer]
33
+ num_slices = 8
34
+
35
+ [model.depformer.transformer]
36
+ d_model = 1024
37
+ num_heads = 16
38
+ num_layers = 6
39
+ dim_feedforward = 4096
40
+ causal = true
41
+ norm_first = true
42
+ bias_ff = false
43
+ bias_attn = false
44
+ context = 32
45
+ max_period = 10000
46
+ use_conv_block = false
47
+ use_conv_bias = true
48
+ gating = "silu"
49
+ norm = "RmsNorm"
50
+ positional_embedding = "None"
51
+ conv_layout = false
52
+ conv_kernel_size = 3
53
+ kv_repeat = 1
54
+ max_seq_len = 4096
55
+
56
+ [model.conditioners.description]
57
+ type = "Lut"
58
+ n_bins = 31
59
+ dim = 16
60
+ possible_values = ["very_bad", "bad", "neutral", "good", "very_good"]