Audio-to-Audio
Safetensors
torch
File size: 3,967 Bytes
d8a2953
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
{
  "encoder_name": "WavLM",
  "encoder_config": {
    "hidden_dims": [
      512,
      512,
      512,
      512,
      512,
      512,
      512
    ],
    "kernel_sizes": [
      10,
      3,
      3,
      3,
      3,
      2,
      2
    ],
    "strides": [
      5,
      2,
      2,
      2,
      2,
      2,
      2
    ],
    "num_layers": 6,
    "dim": 1024,
    "ffn_dim": 4096,
    "num_heads": 16,
    "num_buckets": 320,
    "max_distance": 800,
    "max_cached_steps": 2048,
    "dropout": 0.0,
    "conv_pos": 128,
    "conv_pos_groups": 16,
    "causal": false,
    "window_size": 512,
    "lookahead_size": 3,
    "use_flex_attention": false
  },
  "compressor_name": "FocalEncoder",
  "compressor_config": {
    "input_dim": 1024,
    "output_dim": 32,
    "hidden_dims": [
      1024,
      1024,
      1024
    ],
    "downscale_factors": [
      1,
      1,
      1
    ],
    "focal_window": 14,
    "focal_level": 2,
    "focal_factor": 4,
    "dropout": 0.0,
    "use_post_norm": false,
    "use_layerscale": false,
    "layerscale_init": 0.0001,
    "tanhscale_init": 0.5,
    "normalize_modulator": false,
    "causal": false,
    "window_size": 512
  },
  "boundary_predictor_name": "HazardModel",
  "boundary_predictor_config": {
    "input_dim": 1024,
    "hidden_dims": [
      1024,
      1024,
      1024
    ],
    "downscale_factors": [
      1,
      1,
      1
    ],
    "focal_window": 14,
    "focal_level": 2,
    "focal_factor": 4,
    "dropout": 0.0,
    "use_post_norm": false,
    "use_layerscale": false,
    "layerscale_init": 0.0001,
    "tanhscale_init": 0.5,
    "normalize_modulator": false,
    "causal": false,
    "window_size": 512
  },
  "downsampler_name": "SelectLastPool",
  "downsampler_config": {},
  "quantizer_name": "ScalarSphericalQuantizer",
  "quantizer_config": {
    "dim": 32,
    "n_levels": 4
  },
  "duration_predictor_name": "NegBinModel",
  "duration_predictor_config": {
    "input_dim": 32,
    "hidden_dims": [
      1024,
      1024,
      1024
    ],
    "downscale_factors": [
      1,
      1,
      1
    ],
    "focal_window": 14,
    "focal_level": 2,
    "focal_factor": 4,
    "dropout": 0.0,
    "use_post_norm": false,
    "use_layerscale": false,
    "layerscale_init": 0.0001,
    "tanhscale_init": 0.5,
    "normalize_modulator": false,
    "causal": false,
    "window_size": 512,
    "min_duration": 1,
    "eps": 0.0001
  },
  "upsampler_name": "RepeatInterleaveUnpool",
  "upsampler_config": {},
  "decompressor_name": "FocalDecoder",
  "decompressor_config": {
    "input_dim": 32,
    "output_dim": 1024,
    "hidden_dims": [
      1024,
      1024,
      1024
    ],
    "upscale_factors": [
      1,
      1,
      1
    ],
    "focal_window": 14,
    "focal_level": 2,
    "focal_factor": 4,
    "dropout": 0.0,
    "use_post_norm": false,
    "use_layerscale": false,
    "layerscale_init": 0.0001,
    "tanhscale_init": 0.5,
    "normalize_modulator": false,
    "causal": false,
    "window_size": 512,
    "last_window_size": 512,
    "lookahead_size": 3
  },
  "decoder_name": "Vocos",
  "decoder_config": {
    "input_dim": 1024,
    "num_layers": 8,
    "dim": 512,
    "ffn_dim": 1536,
    "kernel_size": 7,
    "layerscale_init": 0.125,
    "n_fft": 1024,
    "hop_length": 320,
    "causal": false
  },
  "char_aligner_name": "MMS",
  "char_aligner_config": {
    "checkpoint": "facebook/mms-1b-all"
  },
  "retriever_name": "LatentIVF",
  "retriever_config": {
    "input_dim": 1024,
    "latent_dim": 32,
    "hidden_dims": [
      1024,
      1024,
      1024
    ],
    "downscale_factors": [
      1,
      1,
      1
    ],
    "focal_window": 14,
    "focal_level": 2,
    "focal_factor": 4,
    "dropout": 0.0,
    "use_post_norm": false,
    "use_layerscale": false,
    "layerscale_init": 0.0001,
    "tanhscale_init": 0.5,
    "normalize_modulator": false,
    "causal": false,
    "window_size": 512,
    "nlist": 4096,
    "nprobe": 16
  }
}