facebook
/
bart-large-cnn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adding generation config file(s)

This commit is contained in:
Joao Gante 2023-01-24 16:28:55 +00:00 committed by huggingface-web
parent 45b6053e29
commit 3d224934c6
2 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
generation_config.json Normal file
View File

@ -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"
}

16
generation_config_for_summarization.json Normal file
View File

@ -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"
}