diff --git a/config.json b/config.json index 5b0072c..7903079 100644 --- a/config.json +++ b/config.json @@ -48,6 +48,16 @@ "prefix": " ", "pruned_heads": {}, "repetition_penalty": 1.0, + "task_specific_params": { + "summarization": { + "early_stopping": true, + "length_penalty": 2.0, + "max_length": 142, + "min_length": 56, + "no_repeat_ngram_size": 3, + "num_beams": 4 + } + }, "temperature": 1.0, "top_k": 50, "top_p": 1.0,