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

[fix] oogabooga new api support

Browse Source
This commit is contained in:
kwaroran
2023-05-08 22:37:27 +09:00
parent 6d3e292a5b
commit 67208e8dd2
2 changed files with 37 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -109,20 +109,10 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
}
case "textgen_webui":{
let DURL = db.textgenWebUIURL
if((!DURL.endsWith('textgen')) && (!DURL.endsWith('textgen/'))){
if(DURL.endsWith('/')){
DURL += 'run/textgen'
}
else{
DURL += '/run/textgen'
}
}
let bodyTemplate:any
const proompt = stringlizeChat(formated, currentChar.name)
const payload = [
proompt,
{
if(DURL.includes('api')){
bodyTemplate = {
'max_new_tokens': 80,
'do_sample': true,
'temperature': (db.temperature / 100),
@@ -139,14 +129,41 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
'early_stopping': false,
'truncation_length': maxTokens,
'ban_eos_token': false,
'custom_stopping_strings': [`\nUser:`],
'custom_stopping_strings': [`\nUser:`,`\nuser:`],
'seed': -1,
add_bos_token: true,
prompt: proompt
}
];
const bodyTemplate = { "data": [JSON.stringify(payload)] };
}
else{
const payload = [
proompt,
{
'max_new_tokens': 80,
'do_sample': true,
'temperature': (db.temperature / 100),
'top_p': 0.9,
'typical_p': 1,
'repetition_penalty': (db.PresensePenalty / 100),
'encoder_repetition_penalty': 1,
'top_k': 100,
'min_length': 0,
'no_repeat_ngram_size': 0,
'num_beams': 1,
'penalty_alpha': 0,
'length_penalty': 1,
'early_stopping': false,
'truncation_length': maxTokens,
'ban_eos_token': false,
'custom_stopping_strings': [`\nUser:`,`\nuser:`],
'seed': -1,
add_bos_token: true,
}
];
bodyTemplate = { "data": [JSON.stringify(payload)] };
}
const res = await globalFetch(DURL, {
body: bodyTemplate,
headers: {}

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

@@ -11,10 +11,10 @@ export function stringlizeChat(formated:OpenAIChat[], char:string = ''){
resultString.push("'System Note: " + form.content)
}
else if(form.role === 'user'){
resultString.push("User: " + form.content)
resultString.push("user: " + form.content)
}
else if(form.role === 'assistant'){
resultString.push("Assistant: " + form.content)
resultString.push("assistant: " + form.content)
}
}
return resultString.join('\n\n') + `\n\n${char}:`