ybelkada commited on
Commit
9d135aa
1 Parent(s): 9b0ea88

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -11
config.json CHANGED
@@ -28,23 +28,12 @@
28
  "num_hidden_layers": 2,
29
  "pad_token_id": 0,
30
  "qk_layer_norms": false,
31
- "qk_layer_norms_perceiver": false,
32
- "resampler_depth": 2,
33
- "resampler_head_dim": 8,
34
- "resampler_n_heads": 2,
35
- "resampler_n_latents": 16,
36
  "rms_norm_eps": 1e-06,
37
  "tie_word_embeddings": false,
38
  "torch_dtype": "float16",
39
  "transformers_version": "4.27.0.dev0",
40
  "use_cache": true,
41
  "use_resampler": true,
42
- "vision_embed_dim": 32,
43
- "vision_image_size": 30,
44
- "vision_intermediate_size": 37,
45
- "vision_patch_size": 2,
46
- "vision_num_attention_heads": 4,
47
- "vision_num_hidden_layers": 5,
48
  "vocab_size": 32000,
49
  "word_embed_proj_dim": 16,
50
  "vision_config": {
@@ -54,5 +43,12 @@
54
  "patch_size": 2,
55
  "num_attention_heads": 4,
56
  "num_hidden_layers": 5
 
 
 
 
 
 
 
57
  }
58
  }
 
28
  "num_hidden_layers": 2,
29
  "pad_token_id": 0,
30
  "qk_layer_norms": false,
 
 
 
 
 
31
  "rms_norm_eps": 1e-06,
32
  "tie_word_embeddings": false,
33
  "torch_dtype": "float16",
34
  "transformers_version": "4.27.0.dev0",
35
  "use_cache": true,
36
  "use_resampler": true,
 
 
 
 
 
 
37
  "vocab_size": 32000,
38
  "word_embed_proj_dim": 16,
39
  "vision_config": {
 
43
  "patch_size": 2,
44
  "num_attention_heads": 4,
45
  "num_hidden_layers": 5
46
+ },
47
+ "perceiver_config": {
48
+ "qk_layer_norms_perceiver": false,
49
+ "resampler_depth": 2,
50
+ "resampler_head_dim": 8,
51
+ "resampler_n_heads": 2,
52
+ "resampler_n_latents": 16
53
  }
54
  }