Update config.json

config.json

@@ -3,7 +3,7 @@
   "activation_function": "gelu",
   "add_type_embedding": true,
   "architectures": [
-    "OFAForConditionalGeneration"
+    "OFAModel"
   ],
   "attention_dropout": 0.0,
   "attn_scale_factor": 2.0,
@@ -13,20 +13,22 @@
   "code_layernorm_embedding": true,
   "d_model": 768,
   "decoder_attention_heads": 12,
-  "decoder_drop_path_rate": 0.1,
+  "decoder_drop_path_rate": 0.0,
   "decoder_ffn_dim": 3072,
   "decoder_layerdrop": 0.0,
   "decoder_layers": 6,
   "decoder_normalize_before": true,
-  "dropout": 0.0,
+  "decoder_start_token_id": 0,
+  "dropout": 0.1,
   "encoder_attention_heads": 12,
-  "encoder_drop_path_rate": 0.1,
+  "encoder_drop_path_rate": 0.0,
   "encoder_ffn_dim": 3072,
   "encoder_layerdrop": 0.0,
   "encoder_layers": 6,
   "encoder_normalize_before": true,
   "entangle_position_embedding": false,
   "eos_token_id": 2,
+  "forced_eos_token_id": 2,
   "image_bucket_size": 42,
   "init_std": 0.02,
   "is_encoder_decoder": true,
@@ -44,7 +46,7 @@
   "share_decoder_input_output_embed": true,
   "token_bucket_size": 256,
   "torch_dtype": "float32",
-  "transformers_version": "4.18.0.dev0",
-  "use_cache": false,
+  "transformers_version": "4.15.0",
+  "use_cache": true,
   "vocab_size": 59457
 }