Upload BartForConditionalGeneration

config.json

@@ -5,7 +5,7 @@
   "add_bias_logits": false,
   "add_final_layer_norm": false,
   "architectures": [
-    "BartForCausalLM"
+    "BartForConditionalGeneration"
   ],
   "attention_dropout": 0.1,
   "bos_token_id": 101,
@@ -32,8 +32,7 @@
     "2": "LABEL_2"
   },
   "init_std": 0.02,
-  "is_decoder": true,
-  "is_encoder_decoder": false,
+  "is_encoder_decoder": true,
   "label2id": {
     "LABEL_0": 0,
     "LABEL_1": 1,
@@ -46,6 +45,8 @@
   "normalize_embedding": true,
   "num_beams": 4,
   "num_hidden_layers": 6,
+  "output_attentions": true,
+  "output_hidden_states": true,
   "pad_token_id": 0,
   "scale_embedding": false,
   "task_specific_params": {

generation_config.json

@@ -6,6 +6,8 @@
   "forced_eos_token_id": 102,
   "no_repeat_ngram_size": 3,
   "num_beams": 4,
+  "output_attentions": true,
+  "output_hidden_states": true,
   "pad_token_id": 0,
   "transformers_version": "4.28.1"
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7191d4af7646cfba04ffcc66b1e04db369ac38836a3b38379e08c329f84984cd
-size 387558115
+oid sha256:e75788fe0877d4b99b8292fb904aeaf001a74a7200926123a38285ecf7aaacb1
+size 561062835