Create config.json

config.json

@@ -0,0 +1,44 @@
+{
+  "_name_or_path": "TuKoResearch/WavTokenizerSmall",
+  "architectures": [
+    "WavTokenizer"
+  ],
+  "auto_map": {
+    "AutoConfig": "configuration_wavtokenizer.WavTokenizerConfig",
+    "AutoModel": "modeling_wavtokenizer.WavTokenizer"
+  },
+  "model_type": "wavtokenizer",
+  
+  "sample_rate": 24000,
+  "n_fft": 1280,
+  "hop_length": 320,
+  "n_mels": 128,
+  "padding": "center",
+  
+  "feature_dim": 512,
+  "encoder_dim": 64,
+  "encoder_rates": [8, 5, 4, 2],
+  "latent_dim": 512,
+  
+  "codebook_size": 4096,
+  "codebook_dim": 8,
+  "num_quantizers": 1,
+  
+  "backbone_type": "vocos",
+  "backbone_dim": 512,
+  "backbone_num_blocks": 8,
+  "backbone_intermediate_dim": 1536,
+  "backbone_kernel_size": 7,
+  "backbone_layer_scale_init_value": 1e-6,
+  
+  "head_type": "istft",
+  "head_dim": 641,
+  
+  "use_attention": true,
+  "attention_dim": 512,
+  "attention_heads": 8,
+  "attention_layers": 1,
+  
+  "torch_dtype": "float32",
+  "transformers_version": "4.40.0"
+}