tiny/model_cfg.json

{
    "window_length": 512,
    "n_filters": 32,
    "ratios": [
        8,
        4,
        2
    ],
    "kernel_size": 5,
    "last_kernel_size": 5,
    "n_dim": 256,
    "n_head": 4,
    "n_neuro": 16,
    "dropout": 0.0,
    "codebook_dim": 256,
    "codebook_size": 512,
    "num_quantizers": 4,
    "rotation_trick": true,
    "quantize_optimize_method": "ema",
    "overlap_ratio": 0.25,
    "lm_dim": 256,
    "lm_head": 8,
    "lm_depth": 12,
    "lm_dropout": 0.1,
    "mask_ratio": 0.5,
    "num_quantizers_used": 4
}