Update config.json
Browse files- 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 |
}
|