Hugging Face's logo

turkeyju
/
generator_maskgen_kl_l

Safetensors
model_hub_mixin
pytorch_model_hub_mixin
text-to-image-generation
Model card Files Community
generator_maskgen_kl_l
File size: 1,148 Bytes 
bc645b0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
{
    "experiment": {
        "tokenizer_checkpoint": "tatitok_bl32_vae.bin",
        "generator_checkpoint": "maskgen_kl_l.bin"
    },
    "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": 32,
            "scale_factor": 0.7525,
            "finetune_decoder": false,
            "is_legacy": false
        },
        "maskgen": {
            "decoder_embed_dim": 1024,
            "decoder_depth": 16,
            "decoder_num_heads": 16,
            "micro_condition": true,
            "micro_condition_embed_dim": 256,
            "text_drop_prob": 0.1,
            "cfg": 3.0,
            "cfg_schedule": "linear",
            "num_iter": 32,
            "temperature": 1.0,
            "sample_aesthetic_score": 6.5
        }
    },
    "losses": {
        "diffloss_d": 8,
        "diffloss_w": 1024
    },
    "dataset": {
        "preprocessing": {
            "crop_size": 256
        }
    }
}