diff --git a/generation_config.json b/generation_config.json new file mode 100644 index 0000000..4bdb819 --- /dev/null +++ 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" +}