nielsr HF staff commited on
Commit
6a49977
1 Parent(s): 41c0ce2

Push model using huggingface_hub.

Browse files
Files changed (1) hide show
  1. config.json +33 -0
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "experiment": {
3
+ "tokenizer_checkpoint": "tokenizer_titok_l32.bin",
4
+ "generator_checkpoint": "generator_titok_l32.bin"
5
+ },
6
+ "model": {
7
+ "vq_model": {
8
+ "codebook_size": 4096,
9
+ "token_size": 12,
10
+ "use_l2_norm": true,
11
+ "commitment_cost": 0.25,
12
+ "vit_enc_model_size": "large",
13
+ "vit_dec_model_size": "large",
14
+ "vit_enc_patch_size": 16,
15
+ "vit_dec_patch_size": 16,
16
+ "num_latent_tokens": 32
17
+ },
18
+ "generator": {
19
+ "dropout": 0.1,
20
+ "attn_drop": 0.1,
21
+ "num_steps": 8,
22
+ "mask_schedule_strategy": "arccos",
23
+ "class_label_dropout": 0.1,
24
+ "image_seq_len": "${model.vq_model.num_latent_tokens}",
25
+ "condition_num_classes": 1000
26
+ }
27
+ },
28
+ "dataset": {
29
+ "preprocessing": {
30
+ "crop_size": 256
31
+ }
32
+ }
33
+ }