Add customizable ban tokens (#3899)

2023-09-15 14:27:27 -07:00 · 2023-09-15 14:27:27 -07:00 · f01b9aa71f
commit f01b9aa71f
parent fb864dad7b
16 changed files with 56 additions and 5 deletions
--- a/modules/ui.py
+++ b/modules/ui.py
@ -118,6 +118,7 @@ def list_interface_input_elements():
        'guidance_scale',
        'add_bos_token',
        'ban_eos_token',
+        'custom_token_bans',
        'truncation_length',
        'custom_stopping_strings',
        'skip_special_tokens',