Update model_config.yaml
Browse files- model_config.yaml +0 -2
model_config.yaml
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
name: trocr_image2text
|
| 2 |
config_type: model
|
| 3 |
encoder:
|
| 4 |
-
name: trocr_encoder
|
| 5 |
config_type: model
|
| 6 |
hidden_size: 768
|
| 7 |
num_hidden_layers: 12
|
|
@@ -18,7 +17,6 @@ encoder:
|
|
| 18 |
qkv_bias: true
|
| 19 |
encoder_stride: 16
|
| 20 |
decoder:
|
| 21 |
-
name: trocr_decoder
|
| 22 |
config_type: model
|
| 23 |
is_decoder: true
|
| 24 |
add_cross_attention: true
|
|
|
|
| 1 |
name: trocr_image2text
|
| 2 |
config_type: model
|
| 3 |
encoder:
|
|
|
|
| 4 |
config_type: model
|
| 5 |
hidden_size: 768
|
| 6 |
num_hidden_layers: 12
|
|
|
|
| 17 |
qkv_bias: true
|
| 18 |
encoder_stride: 16
|
| 19 |
decoder:
|
|
|
|
| 20 |
config_type: model
|
| 21 |
is_decoder: true
|
| 22 |
add_cross_attention: true
|