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

[feat] add novellist damsel

Browse Source
This commit is contained in:
kwaroran
2023-06-26 00:25:25 +09:00
parent 90b60d71b5
commit 0bbd5dbdac
2 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/lib/UI/ModelList.svelte
View File

@@ -37,6 +37,10 @@
return "Plugin" return "Plugin"
case "novelai": case "novelai":
return "NovelAI" return "NovelAI"
case "novellist":
return "NovelList SuperTrin"
case "novellist damsel":
return "NovelList Damsel"
default: default:
if(name.startsWith("horde:::")){ if(name.startsWith("horde:::")){
return name.replace(":::", " ") return name.replace(":::", " ")
@@ -79,8 +83,10 @@
<button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('textgen_webui')}}>Oobabooga WebUI</button> <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('textgen_webui')}}>Oobabooga WebUI</button>
<button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('palm2')}}>Google PaLM2</button> <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('palm2')}}>Google PaLM2</button>
<button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('kobold')}}>Kobold</button> <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('kobold')}}>Kobold</button>
<button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('novellist')}}>Novellist</button> <Arcodion name="Novellist">
<button class="p-2 hover:text-green-500" on:click={() => {changeModel('novellist')}}>SuperTrin</button>
<button class="p-2 hover:text-green-500" on:click={() => {changeModel('novellist_damsel')}}>Damsel</button>
</Arcodion>
{#if isTauri ||isNodeServer} {#if isTauri ||isNodeServer}
<button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('novelai')}}>NovelAI Clio</button> <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('novelai')}}>NovelAI Clio</button>
<button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('deepai')}}>DeepAI</button> <button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('deepai')}}>DeepAI</button>

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

@@ -556,7 +556,8 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
result: data.results[0].text result: data.results[0].text
} }
} }
case "novellist":{ case "novellist":
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/';
@@ -572,6 +573,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
top_p: 0.7, top_p: 0.7,
tailfree: 1.0, tailfree: 1.0,
rep_pen: arg.frequencyPenalty ?? (db.frequencyPenalty / 100) + 1, rep_pen: arg.frequencyPenalty ?? (db.frequencyPenalty / 100) + 1,
model: aiModel === 'novellist_damsel' ? 'damsel' : 'supertrin'
}; };
const response = await globalFetch(api_server_url + '/api', { const response = await globalFetch(api_server_url + '/api', {