Skip to content

Commit b5931e7

Browse files
committed
Added more tags to break from when encountered during generation
1 parent 36f9308 commit b5931e7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def chat():
141141

142142

143143
# Check for any of the tags
144-
tags = ["<assistant>", "<human>", "<npc>", "<system>"]
144+
tags = ["<assistant>", "<human>", "<npc>", "<system>", "</assistant>", "</human>", "</npc>", "</system>"]
145145
for tag in tags:
146146
if tag in lower_generated_text:
147147
print(f"Found tag: {tag}")

0 commit comments

Comments
 (0)