From ceaa69c7bc5e512b5007106a7ccbb7daf24b2c79 Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Tue, 24 Jan 2023 14:50:31 +0000 Subject: [PATCH] Adding generation config file(s) --- generation_config.json | 7 +++++++ generation_config_for_text_generation.json | 9 +++++++++ 2 files changed, 16 insertions(+) create mode 100644 generation_config.json create mode 100644 generation_config_for_text_generation.json diff --git a/generation_config.json b/generation_config.json new file mode 100644 index 0000000..d7822b8 --- /dev/null +++ b/generation_config.json @@ -0,0 +1,7 @@ +{ + "_from_model_config": true, + "bos_token_id": 1, + "eos_token_id": 2, + "pad_token_id": 5, + "transformers_version": "4.27.0.dev0" +} diff --git a/generation_config_for_text_generation.json b/generation_config_for_text_generation.json new file mode 100644 index 0000000..32e10ba --- /dev/null +++ b/generation_config_for_text_generation.json @@ -0,0 +1,9 @@ +{ + "_from_model_config": true, + "bos_token_id": 1, + "do_sample": true, + "eos_token_id": 2, + "max_length": 250, + "pad_token_id": 5, + "transformers_version": "4.27.0.dev0" +}