Create config.json

config.json

@@ -0,0 +1,63 @@
+{
+  "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,
+    "dropout": 0.0,
+    "conv_pos": 128,
+    "conv_pos_groups": 16
+  },
+  "compressor_name": "FocalEncoder",
+  "compressor_config": {
+    "input_dim": 1024,
+    "output_dim": 13,
+    "hidden_dims": [1024, 512, 256],
+    "downscale_factors": [2, 1, 1],
+    "focal_window": 7,
+    "focal_level": 2,
+    "focal_factor": 2,
+    "dropout": 0.0,
+    "use_post_norm": false,
+    "use_layerscale": false,
+    "layerscale_init": 0.0001,
+    "normalize_modulator": false
+  },
+  "quantizer_name": "BinarySphericalQuantizer",
+  "quantizer_config": {
+    "codebook_size": 8192
+  },
+  "decompressor_name": "FocalDecoder",
+  "decompressor_config": {
+    "input_dim": 13,
+    "output_dim": 1024,
+    "hidden_dims": [256, 512, 1024],
+    "upscale_factors": [1, 1, 2],
+    "focal_window": 7,
+    "focal_level": 2,
+    "focal_factor": 2,
+    "dropout": 0.0,
+    "use_post_norm": false,
+    "use_layerscale": false,
+    "layerscale_init": 0.0001,
+    "normalize_modulator": false
+  },
+  "decoder_name": "Vocos",
+  "decoder_config": {
+    "input_channels": 1024,
+    "num_layers": 8,
+    "dim": 512,
+    "ffn_dim": 1536,
+    "kernel_size": 7,
+    "padding": 3,
+    "layerscale_init": null,
+    "n_fft": 1024,
+    "hop_length": 98
+  }
+}