config.json

{
    "experiment": {
        "tokenizer_checkpoint": "titok_bl128_vae_c16.bin",
        "output_dir": "titok_bl128_vae_c16"
    },
    "model": {
        "vq_model": {
            "quantize_mode": "vae",
            "token_size": 16,
            "vit_enc_model_size": "base",
            "vit_dec_model_size": "large",
            "vit_enc_patch_size": 16,
            "vit_dec_patch_size": 16,
            "num_latent_tokens": 128,
            "finetune_decoder": false,
            "is_legacy": false
        }
    },
    "dataset": {
        "preprocessing": {
            "crop_size": 256
        }
    }
}