mirror of
https://github.com/oobabooga/textgen.git
synced 2026-08-03 16:50:46 -05:00
Add customizable ban tokens (#3899)
This commit is contained in:
@@ -45,6 +45,7 @@ def run(prompt):
|
||||
'add_bos_token': True,
|
||||
'truncation_length': 2048,
|
||||
'ban_eos_token': False,
|
||||
'custom_token_bans': '',
|
||||
'skip_special_tokens': True,
|
||||
'stopping_strings': []
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user