Update config.json

config.json

@@ -102,20 +102,12 @@
   },
   "max_position_embeddings": 1024,
   "model_type": "bart",
-  "n_ctx": 1024,
-  "n_embd": 1024,
-  "n_head": 16,
-  "n_inner": null,
-  "n_layer": 24,
-  "n_positions": 1024,
-  "no_repeat_ngram_size": 2,
-  "num_beams": 5,
-  "num_embeddings": 50257,
-  "num_heads": 64,
-  "num_layers": 24,
-  "output_attentions": true,
-  "output_hidden_states": true,
-  "pad_token_id": 50267,
+  "no_repeat_ngram_size": 3,
+  "normalize_before": false,
+  "num_beams": 4,
+  "num_hidden_layers": 12,
+  "pad_token_id": 1,
+  "scale_embedding": false,
   "reorder_and_upcast_attn": false,
   "resid_pdrop": 0.1,
   "scale_attn_by_inverse_layer_idx": false,
@@ -157,10 +149,28 @@
       "temperature": 0.9,
       "top_k": 50,
       "top_p": 0.95
+    },
+    "summarization": {
+      "length_penalty": 1.0,
+      "max_length": 128,
+      "min_length": 12,
+      "num_beams": 4
+    },
+    "summarization_cnn": {
+      "length_penalty": 2.0,
+      "max_length": 142,
+      "min_length": 56,
+      "num_beams": 4
+    },
+    "summarization_xsum": {
+      "length_penalty": 1.0,
+      "max_length": 62,
+      "min_length": 11,
+      "num_beams": 6
     }
   },
   "torch_dtype": "float32",
-  "transformers_version": "4.44.0",
+  "transformers_version": "4.7.0.dev0",
   "use_cache": true,
-  "vocab_size": 50257
-}
+  "vocab_size": 50265
+}