File size: 467 Bytes
51ce47d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
experiment:
    tokenizer_checkpoint: "titok_bl128_vae_c16.bin"
    output_dir: "titok_bl128_vae_c16"
model:
    vq_model:
        quantize_mode: "vae"
        token_size: 16 
        # vit arch
        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