mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[fix] autosuggest stop strings

Browse Source
This commit is contained in:
aegkmq
2023-08-03 15:19:56 +09:00
parent 4ca304d26d
commit 2308e478ab
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/ts/process/stringlize.ts
View File

@@ -84,9 +84,7 @@ export function getStopStrings(suggesting:boolean=false){
"<|end", "<|end",
"<|im_end", "<|im_end",
userPrefix, userPrefix,
`*You '`, "\nYou ",
`\nYou '`,
` You '`,
`*${username}'`, `*${username}'`,
`*${username} `, `*${username} `,
`\n${username} `, `\n${username} `,
@@ -98,6 +96,10 @@ export function getStopStrings(suggesting:boolean=false){
if(suggesting){ if(suggesting){
stopStrings.push("\n\n") stopStrings.push("\n\n")
} }
if(!suggesting){
stopStrings.push("*You ")
stopStrings.push(" You ")
}
for (const user of userStrings){ for (const user of userStrings){
for (const u of [ for (const u of [
user.toLowerCase(), user.toLowerCase(),