Fix split.

jondurbin · Aug 23, 2023 · cd29c71 · cd29c71
1 parent ae6486d
commit cd29c71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/airoboros/lmoe/api.py b/airoboros/lmoe/api.py
@@ -194,7 +194,7 @@ def complete_request(request):
             skip_special_tokens=True,
             clean_up_tokenization_spaces=False,
         )[0]
-        .split("ASSISTANT:")[1]
+        .split("ASSISTANT:")[-1]
         .strip()
     )
     response = re.sub(r"[\s\n]*(USER|ASSISTANT):\s*$", "", response, re.DOTALL)