rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Adventure stoppers = regex

Browse Source
This commit is contained in:
Henk
2023-07-23 20:45:48 +02:00
parent d70481874c
commit e33a58b74a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modeling/stoppers.py
View File

@@ -127,7 +127,7 @@ class Stoppers:
if "completed" not in model.gen_state: if "completed" not in model.gen_state:
model.gen_state["completed"] = [False] * len(input_ids) model.gen_state["completed"] = [False] * len(input_ids)
if utils.koboldai_vars.adventure: if utils.koboldai_vars.adventure:
extra_options = ["> You", "You:", "\n\n You", "\n\nYou", ". You"] extra_options = [">", "\n>"]
for option in extra_options: for option in extra_options:
if option not in utils.koboldai_vars.stop_sequence: if option not in utils.koboldai_vars.stop_sequence:
utils.koboldai_vars.stop_sequence.append(option) utils.koboldai_vars.stop_sequence.append(option)