Adding generation config file(s)

2023-01-24 16:29:06 +00:00 · 2023-01-24 16:29:06 +00:00 · a0d0ef7fe2
parent c468b2376f
commit a0d0ef7fe2
1 changed files with 15 additions and 0 deletions
--- a/generation_config.json
+++ b/generation_config.json
@ -0,0 +1,15 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1,
+  "decoder_start_token_id": 1,
+  "encoder_no_repeat_ngram_size": 3,
+  "eos_token_id": 2,
+  "forced_eos_token_id": 2,
+  "length_penalty": 0.65,
+  "max_length": 60,
+  "min_length": 20,
+  "no_repeat_ngram_size": 3,
+  "num_beams": 10,
+  "pad_token_id": 0,
+  "transformers_version": "4.27.0.dev0"
+}