Two new options: truncation length and ban eos token

This commit is contained in:
oobabooga
2023-04-11 18:46:06 -03:00
committed by GitHub
parent 749c08a4ff
commit cacbcda208
6 changed files with 62 additions and 48 deletions

View File

@@ -37,6 +37,10 @@ settings = {
'custom_stopping_strings': '',
'stop_at_newline': False,
'add_bos_token': True,
'ban_eos_token': False,
'truncation_length': 2048,
'truncation_length_min': 0,
'truncation_length_max': 4096,
'chat_prompt_size': 2048,
'chat_prompt_size_min': 0,
'chat_prompt_size_max': 2048,