From 9e4cf98cfaef428b8e374a4dfb824b3696c34f83 Mon Sep 17 00:00:00 2001 From: Joao Gante Date: Tue, 24 Jan 2023 13:52:18 +0000 Subject: [PATCH] Adding generation config file(s) --- generation_config.json | 6 ++++++ {pipeline}_generation_config.json | 8 ++++++++ 2 files changed, 14 insertions(+) create mode 100644 generation_config.json create mode 100644 {pipeline}_generation_config.json diff --git a/generation_config.json b/generation_config.json new file mode 100644 index 0000000..057e430 --- /dev/null +++ b/generation_config.json @@ -0,0 +1,6 @@ +{ + "_from_model_config": true, + "bos_token_id": 50256, + "eos_token_id": 50256, + "transformers_version": "4.27.0.dev0" +} diff --git a/{pipeline}_generation_config.json b/{pipeline}_generation_config.json new file mode 100644 index 0000000..c791c4d --- /dev/null +++ b/{pipeline}_generation_config.json @@ -0,0 +1,8 @@ +{ + "_from_model_config": true, + "bos_token_id": 50256, + "do_sample": true, + "eos_token_id": 50256, + "max_length": 50, + "transformers_version": "4.27.0.dev0" +}