arxyzan commited on
Commit
1c8a07a
1 Parent(s): 9dbadf4

Hezar: Upload model_config.yaml

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