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

[fix] oobabooga prompts

Browse Source
This commit is contained in:
aegkmq
2023-08-01 15:58:41 +09:00
parent 5ef6f00a8c
commit 8aacea9f1a
3 changed files with 58 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/ts/process/request.ts
View File

@@ -3,7 +3,7 @@ import type { OpenAIChat, OpenAIChatFull } from ".";
import { DataBase, setDatabase, type character } from "../storage/database";
import { pluginProcess } from "../plugins/plugins";
import { language } from "../../lang";
import { stringlizeAINChat, stringlizeChat, stringlizeChatOba, unstringlizeAIN, unstringlizeChat } from "./stringlize";
import { stringlizeAINChat, stringlizeChat, stringlizeChatOba, getStopStrings, unstringlizeAIN, unstringlizeChat } from "./stringlize";
import { globalFetch, isNodeServer, isTauri } from "../storage/globalApi";
import { sleep } from "../util";
import { createDeep } from "./deepai";
@@ -375,11 +375,12 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
let DURL = db.textgenWebUIURL
let bodyTemplate:any
const proompt = stringlizeChatOba(formated, currentChar?.name ?? '')
const stopStrings = getStopStrings()
if(!DURL.endsWith('generate')){
DURL = DURL + "/v1/generate"
}
const stopStrings = [`\nUser:`,`\nuser:`,`\n${db.username}:`]
console.log(proompt)
console.log(stopStrings)
bodyTemplate = {
'max_new_tokens': db.maxResponse,
'do_sample': true,