stas commited on
Commit
44215ea
1 Parent(s): 5312415

Update config.json (#5)

Browse files

- Update config.json (8cc3f584b9a160ae479de202e870c7199626d3d9)

Files changed (1) hide show
  1. config.json +17 -13
config.json CHANGED
@@ -27,22 +27,26 @@
27
  "num_hidden_layers": 80,
28
  "pad_token_id": 0,
29
  "qk_layer_norms": true,
30
- "qk_layer_norms_perceiver": true,
31
- "resampler_depth": 6,
32
- "resampler_head_dim": 96,
33
- "resampler_n_heads": 16,
34
- "resampler_n_latents": 64,
35
  "rms_norm_eps": 1e-05,
36
  "tie_word_embeddings": false,
37
  "torch_dtype": "bfloat16",
38
  "transformers_version": "4.28.0.dev0",
39
  "use_cache": true,
40
  "use_resampler": true,
41
- "vision_embed_dim": 1280,
42
- "vision_image_size": 224,
43
- "vision_intermediate_size": 5120,
44
- "vision_patch_size": 14,
45
- "vision_num_attention_heads": 16,
46
- "vision_num_hidden_layers": 32,
47
- "vocab_size": 32000
48
- }
 
 
 
 
 
 
 
 
 
 
27
  "num_hidden_layers": 80,
28
  "pad_token_id": 0,
29
  "qk_layer_norms": true,
 
 
 
 
 
30
  "rms_norm_eps": 1e-05,
31
  "tie_word_embeddings": false,
32
  "torch_dtype": "bfloat16",
33
  "transformers_version": "4.28.0.dev0",
34
  "use_cache": true,
35
  "use_resampler": true,
36
+ "vocab_size": 32000,
37
+ "vision_config": {
38
+ "embed_dim": 1280,
39
+ "image_size": 224,
40
+ "intermediate_size": 5120,
41
+ "patch_size": 14,
42
+ "num_attention_heads": 16,
43
+ "num_hidden_layers": 32
44
+ },
45
+ "perceiver_config": {
46
+ "qk_layer_norms_perceiver": true,
47
+ "resampler_depth": 6,
48
+ "resampler_head_dim": 96,
49
+ "resampler_n_heads": 16,
50
+ "resampler_n_latents": 64
51
+ }
52
+ }