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

modify adv stopper

Browse Source
This commit is contained in:
YellowRoseCx
2023-06-15 17:15:33 -05:00
committed by GitHub
parent 877028ec7f
commit 83493dff2e
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

@@ -129,7 +129,7 @@ class Stoppers:
model.gen_state["completed"] = [False] * len(input_ids) model.gen_state["completed"] = [False] * len(input_ids)
for i in range(len(input_ids)): for i in range(len(input_ids)):
if (data[i][-6:] == " > You"): if (data[i][-6:] == "> You " or data[i][-4:] == "You:"):
model.gen_state["completed"][i] = True model.gen_state["completed"][i] = True
if all(model.gen_state["completed"]): if all(model.gen_state["completed"]):