diff --git a/src/ts/process/stringlize.ts b/src/ts/process/stringlize.ts index 16aee013..b8de559c 100644 --- a/src/ts/process/stringlize.ts +++ b/src/ts/process/stringlize.ts @@ -84,9 +84,7 @@ export function getStopStrings(suggesting:boolean=false){ "<|end", "<|im_end", userPrefix, - `*You '`, - `\nYou '`, - ` You '`, + "\nYou ", `*${username}'`, `*${username} `, `\n${username} `, @@ -98,6 +96,10 @@ export function getStopStrings(suggesting:boolean=false){ if(suggesting){ stopStrings.push("\n\n") } + if(!suggesting){ + stopStrings.push("*You ") + stopStrings.push(" You ") + } for (const user of userStrings){ for (const u of [ user.toLowerCase(),