turkeyju commited on
Commit
bc645b0
·
verified ·
1 Parent(s): de47abd

Push model using huggingface_hub.

Browse files
Files changed (1) hide show
  1. config.json +42 -0
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "experiment": {
3
+ "tokenizer_checkpoint": "tatitok_bl32_vae.bin",
4
+ "generator_checkpoint": "maskgen_kl_l.bin"
5
+ },
6
+ "model": {
7
+ "vq_model": {
8
+ "quantize_mode": "vae",
9
+ "token_size": 16,
10
+ "vit_enc_model_size": "base",
11
+ "vit_dec_model_size": "large",
12
+ "vit_enc_patch_size": 16,
13
+ "vit_dec_patch_size": 16,
14
+ "num_latent_tokens": 32,
15
+ "scale_factor": 0.7525,
16
+ "finetune_decoder": false,
17
+ "is_legacy": false
18
+ },
19
+ "maskgen": {
20
+ "decoder_embed_dim": 1024,
21
+ "decoder_depth": 16,
22
+ "decoder_num_heads": 16,
23
+ "micro_condition": true,
24
+ "micro_condition_embed_dim": 256,
25
+ "text_drop_prob": 0.1,
26
+ "cfg": 3.0,
27
+ "cfg_schedule": "linear",
28
+ "num_iter": 32,
29
+ "temperature": 1.0,
30
+ "sample_aesthetic_score": 6.5
31
+ }
32
+ },
33
+ "losses": {
34
+ "diffloss_d": 8,
35
+ "diffloss_w": 1024
36
+ },
37
+ "dataset": {
38
+ "preprocessing": {
39
+ "crop_size": 256
40
+ }
41
+ }
42
+ }