Hezar: Upload model_config.yaml
Browse files- model_config.yaml +0 -4
model_config.yaml
CHANGED
|
@@ -1,8 +1,6 @@
|
|
| 1 |
name: vit_gpt2_image2text
|
| 2 |
config_type: model
|
| 3 |
encoder:
|
| 4 |
-
name: vit_gpt2_encoder
|
| 5 |
-
config_type: model
|
| 6 |
hidden_size: 768
|
| 7 |
num_hidden_layers: 12
|
| 8 |
num_attention_heads: 12
|
|
@@ -18,8 +16,6 @@ encoder:
|
|
| 18 |
qkv_bias: true
|
| 19 |
encoder_stride: 16
|
| 20 |
decoder:
|
| 21 |
-
name: vit_gpt2_decoder
|
| 22 |
-
config_type: model
|
| 23 |
add_cross_attention: true
|
| 24 |
vocab_size: 42001
|
| 25 |
attn_pdrop: 0.1
|
|
|
|
| 1 |
name: vit_gpt2_image2text
|
| 2 |
config_type: model
|
| 3 |
encoder:
|
|
|
|
|
|
|
| 4 |
hidden_size: 768
|
| 5 |
num_hidden_layers: 12
|
| 6 |
num_attention_heads: 12
|
|
|
|
| 16 |
qkv_bias: true
|
| 17 |
encoder_stride: 16
|
| 18 |
decoder:
|
|
|
|
|
|
|
| 19 |
add_cross_attention: true
|
| 20 |
vocab_size: 42001
|
| 21 |
attn_pdrop: 0.1
|