valhalla commited on
Commit
f6f42b8
1 Parent(s): 9662f5b

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. config.json +39 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "MaskGiTUViT_v2",
3
+ "_version": "0.0.1",
4
+ "add_cond_embeds": true,
5
+ "add_micro_cond_embeds": true,
6
+ "attention_dropout": 0.0,
7
+ "block_num_heads": 12,
8
+ "block_out_channels": [
9
+ 768
10
+ ],
11
+ "codebook_size": 8192,
12
+ "cond_embed_dim": 768,
13
+ "decay": 0.9999,
14
+ "encoder_hidden_size": 768,
15
+ "force_down_up_sample": true,
16
+ "hidden_dropout": 0.0,
17
+ "hidden_size": 1024,
18
+ "in_channels": 768,
19
+ "intermediate_size": 2816,
20
+ "inv_gamma": 1.0,
21
+ "layer_norm_eps": 1e-06,
22
+ "ln_elementwise_affine": true,
23
+ "mask_token_id": 8255,
24
+ "micro_cond_embed_dim": 1280,
25
+ "micro_cond_encode_dim": 256,
26
+ "min_decay": 0.0,
27
+ "norm_type": "rmsnorm",
28
+ "num_attention_heads": 16,
29
+ "num_hidden_layers": 22,
30
+ "num_res_blocks": 3,
31
+ "optimization_step": 2500,
32
+ "power": 0.6666666666666666,
33
+ "update_after_step": 0,
34
+ "use_bias": false,
35
+ "use_ema_warmup": false,
36
+ "use_fused_mlp": false,
37
+ "use_fused_residual_norm": true,
38
+ "vocab_size": 8256
39
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:52aae0e7982871c545f69218f9f857ff2efb0f8e289ccbde9cebe90dcd200de3
3
+ size 2433247453