From 83493dff2e7d76fad008dad3cfe8b6c400134d34 Mon Sep 17 00:00:00 2001
From: YellowRoseCx <80486540+YellowRoseCx@users.noreply.github.com>
Date: Thu, 15 Jun 2023 17:15:33 -0500
Subject: [PATCH] modify adv stopper

---
 modeling/stoppers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modeling/stoppers.py b/modeling/stoppers.py
index 0fc6ce10..8fe5c229 100644
--- a/modeling/stoppers.py
+++ b/modeling/stoppers.py
@@ -129,7 +129,7 @@ class Stoppers:
             model.gen_state["completed"] = [False] * 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
                 
         if all(model.gen_state["completed"]):