diff --git a/config.json b/config.json index cd1f7be..683b1a9 100644 --- a/config.json +++ b/config.json @@ -16,7 +16,9 @@ "encoder_layerdrop": 0.0, "encoder_layers": 12, "eos_token_id": 2, - "eos_token_ids": 0, + "eos_token_ids": [ + 2 + ], "finetuning_task": null, "id2label": { "0": "LABEL_0", @@ -40,7 +42,7 @@ "num_return_sequences": 1, "output_attentions": false, "output_hidden_states": false, - "output_past": false, + "output_past": true, "pad_token_id": 1, "pruned_heads": {}, "repetition_penalty": 1.0,