Adding generation config file(s)
This commit is contained in:
parent
45b6053e29
commit
3d224934c6
16
generation_config.json
Normal file
16
generation_config.json
Normal 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
16
generation_config_for_summarization.json
Normal 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"
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user