Update config.json
Browse files- config.json +4 -4
config.json
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
{
|
| 2 |
-
"_name_or_path": "TuKoResearch/
|
| 3 |
"architectures": [
|
| 4 |
"WavTokenizer"
|
| 5 |
],
|
|
@@ -25,9 +25,9 @@
|
|
| 25 |
"num_quantizers": 1,
|
| 26 |
|
| 27 |
"backbone_type": "vocos",
|
| 28 |
-
"backbone_dim":
|
| 29 |
"backbone_num_blocks": 8,
|
| 30 |
-
"backbone_intermediate_dim":
|
| 31 |
"backbone_kernel_size": 7,
|
| 32 |
"backbone_layer_scale_init_value": 1e-6,
|
| 33 |
|
|
@@ -35,7 +35,7 @@
|
|
| 35 |
"head_dim": 641,
|
| 36 |
|
| 37 |
"use_attention": true,
|
| 38 |
-
"attention_dim":
|
| 39 |
"attention_heads": 8,
|
| 40 |
"attention_layers": 1,
|
| 41 |
|
|
|
|
| 1 |
{
|
| 2 |
+
"_name_or_path": "TuKoResearch/WavTokenizer",
|
| 3 |
"architectures": [
|
| 4 |
"WavTokenizer"
|
| 5 |
],
|
|
|
|
| 25 |
"num_quantizers": 1,
|
| 26 |
|
| 27 |
"backbone_type": "vocos",
|
| 28 |
+
"backbone_dim": 768,
|
| 29 |
"backbone_num_blocks": 8,
|
| 30 |
+
"backbone_intermediate_dim": 2304,
|
| 31 |
"backbone_kernel_size": 7,
|
| 32 |
"backbone_layer_scale_init_value": 1e-6,
|
| 33 |
|
|
|
|
| 35 |
"head_dim": 641,
|
| 36 |
|
| 37 |
"use_attention": true,
|
| 38 |
+
"attention_dim": 768,
|
| 39 |
"attention_heads": 8,
|
| 40 |
"attention_layers": 1,
|
| 41 |
|