Hezar: Upload model_config.yaml

model_config.yaml

@@ -0,0 +1,53 @@
+name: vit_roberta_image2text
+config_type: model
+encoder:
+  name: null
+  config_type: model
+  hidden_size: 768
+  num_hidden_layers: 12
+  num_attention_heads: 12
+  intermediate_size: 3072
+  hidden_act: gelu
+  hidden_dropout_prob: 0.0
+  attention_probs_dropout_prob: 0.0
+  initializer_range: 0.02
+  layer_norm_eps: 1.0e-12
+  image_size: 224
+  patch_size: 16
+  num_channels: 3
+  qkv_bias: true
+  encoder_stride: 16
+decoder:
+  name: null
+  config_type: model
+  is_decoder: true
+  add_cross_attention: true
+  attention_probs_dropout_prob: 0.1
+  bos_token_id: 0
+  eos_token_id: 2
+  classifier_dropout: null
+  gradient_checkpointing: false
+  hidden_act: gelu
+  hidden_dropout_prob: 0.1
+  hidden_size: 768
+  initializer_range: 0.02
+  intermediate_size: 3072
+  layer_norm_eps: 1.0e-12
+  max_position_embeddings: 514
+  num_attention_heads: 12
+  num_hidden_layers: 12
+  pad_token_id: 1
+  position_embedding_type: absolute
+  type_vocab_size: 1
+  use_cache: true
+  vocab_size: 42000
+generation:
+  bos_token_id: 0
+  decoder_start_token_id: 0
+  early_stopping: true
+  eos_token_id: 2
+  length_penalty: 2.0
+  max_length: 64
+  no_repeat_ngram_size: 3
+  num_beams: 4
+  pad_token_id: 1