Fixes

config.json

@@ -3,39 +3,18 @@
     "GPT2LMHeadModel"
   ],
   "attn_pdrop": 0.0,
-  "bos_token_id": null,
-  "do_sample": false,
+  "bos_token_id": 2,
   "embd_pdrop": 0.0,
-  "eos_token_ids": null,
-  "finetuning_task": null,
-  "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1"
-  },
+  "eos_token_id": 2,
   "initializer_range": 0.02,
-  "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1
-  },
   "layer_norm_epsilon": 1e-06,
-  "length_penalty": 1.0,
-  "max_length": 20,
   "model_type": "gpt2",
   "n_ctx": 512,
   "n_embd": 1280,
   "n_head": 20,
   "n_layer": 36,
   "n_positions": 512,
-  "num_beams": 5,
-  "num_labels": 2,
-  "num_return_sequences": 5,
-  "output_attentions": false,
-  "output_hidden_states": false,
-  "output_past": true,
   "pad_token_id": 1,
-  "prefix": "</s>",
-  "pruned_heads": {},
-  "repetition_penalty": 1.0,
   "resid_pdrop": 0.0,
   "summary_activation": null,
   "summary_first_dropout": 0.0,
@@ -43,9 +22,12 @@
   "summary_type": "cls_index",
   "summary_use_proj": true,
   "temperature": 1.0,
-  "top_k": 50,
-  "top_p": 1.0,
-  "torchscript": false,
-  "use_bfloat16": false,
-  "vocab_size": 50261
+  "task_specific_params": {
+    "text-generation": {
+      "do_sample": true,
+      "max_length": 50
+    }
+  },
+  "vocab_size": 50261,
+  "prefix": "</s>"
 }