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

[feat] ain bias

Browse Source
This commit is contained in:
kwaroran
2023-08-04 21:05:12 +09:00
parent 4aefd681fe
commit 38195761f8
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -685,7 +685,13 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
case "novellist_damsel":{ case "novellist_damsel":{
const auth_key = db.novellistAPI; const auth_key = db.novellistAPI;
const api_server_url = 'https://api.tringpt.com/'; const api_server_url = 'https://api.tringpt.com/';
const logit_bias:string[] = []
const logit_bias_values:string[] = []
for(let i=0;i<biasString.length;i++){
const bia = biasString[i]
logit_bias.push(bia[0])
logit_bias_values.push(bia[1].toString())
}
const headers = { const headers = {
'Authorization': `Bearer ${auth_key}`, 'Authorization': `Bearer ${auth_key}`,
'Content-Type': 'application/json' 'Content-Type': 'application/json'
@@ -705,6 +711,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
badwords: db.ainconfig.badwords, badwords: db.ainconfig.badwords,
model: aiModel === 'novellist_damsel' ? 'damsel' : 'supertrin', model: aiModel === 'novellist_damsel' ? 'damsel' : 'supertrin',
stoptokens: ["「"].join("<<|>>") + db.ainconfig.stoptokens, stoptokens: ["「"].join("<<|>>") + db.ainconfig.stoptokens,
logit_bias: (logit_bias.length > 0) ? logit_bias.join("<<|>>") : undefined,
logit_bias_values: (logit_bias_values.length > 0) ? logit_bias_values.join("|") : undefined,
}; };
const response = await globalFetch(api_server_url + '/api', { const response = await globalFetch(api_server_url + '/api', {
method: 'POST', method: 'POST',