arxyzan commited on
Commit
4baf369
1 Parent(s): 7149eca

Update model_config.yaml

Browse files
Files changed (1) hide show
  1. 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