From 3d224934c6541b2b9147e023c2f6f6fe49bd27e1 Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Tue, 24 Jan 2023 16:28:55 +0000 Subject: [PATCH] Adding generation config file(s) --- generation_config.json | 16 ++++++++++++++++ generation_config_for_summarization.json | 16 ++++++++++++++++ 2 files changed, 32 insertions(+) create mode 100644 generation_config.json create mode 100644 generation_config_for_summarization.json diff --git a/generation_config.json b/generation_config.json new file mode 100644 index 0000000..f729590 --- /dev/null +++ b/generation_config.json @@ -0,0 +1,16 @@ +{ + "_from_model_config": true, + "bos_token_id": 0, + "decoder_start_token_id": 2, + "early_stopping": true, + "eos_token_id": 2, + "forced_bos_token_id": 0, + "forced_eos_token_id": 2, + "length_penalty": 2.0, + "max_length": 142, + "min_length": 56, + "no_repeat_ngram_size": 3, + "num_beams": 4, + "pad_token_id": 1, + "transformers_version": "4.27.0.dev0" +} diff --git a/generation_config_for_summarization.json b/generation_config_for_summarization.json new file mode 100644 index 0000000..f729590 --- /dev/null +++ b/generation_config_for_summarization.json @@ -0,0 +1,16 @@ +{ + "_from_model_config": true, + "bos_token_id": 0, + "decoder_start_token_id": 2, + "early_stopping": true, + "eos_token_id": 2, + "forced_bos_token_id": 0, + "forced_eos_token_id": 2, + "length_penalty": 2.0, + "max_length": 142, + "min_length": 56, + "no_repeat_ngram_size": 3, + "num_beams": 4, + "pad_token_id": 1, + "transformers_version": "4.27.0.dev0" +}