lfq_tokenizer_B_256_ema/config.json

{
    "auto_map": {
        "AutoModel": "modeling_lfq_tokenizer.LFQTokenizer",
        "AutoConfig": "configuration_lfq_tokenizer.LFQTokenizerConfig"
    },
    "model_type": "lfq_tokenizer",
    "encoder_decoder_config": {
       "ch": 128,
       "in_channels": 3,
       "out_ch": 3,
       "z_channels": 18,
       "num_res_blocks": 2,
       "ch_mult": [1, 1, 2, 2, 4]
    },
    "quantizer_config": {
        "dim": 18,
        "codebook_size": 262144,
        "batch_maximization_weight": 1.0,
        "sample_minimization_weight": 1.0
    }
}