[feat] added novelai

This commit is contained in:
kwaroran
2023-05-29 15:16:06 +09:00
parent 3102e09cd5
commit d60c916feb
2 changed files with 7 additions and 1 deletions

View File

@@ -9,7 +9,7 @@
import ModelList from "src/lib/UI/ModelList.svelte"; import ModelList from "src/lib/UI/ModelList.svelte";
import DropList from "src/lib/SideBars/DropList.svelte"; import DropList from "src/lib/SideBars/DropList.svelte";
import { PlusIcon, TrashIcon } from "lucide-svelte"; import { PlusIcon, TrashIcon } from "lucide-svelte";
import { onDestroy } from "svelte"; import { onDestroy } from "svelte";
let tokens = { let tokens = {
mainPrompt: 0, mainPrompt: 0,
jailbreak: 0, jailbreak: 0,

View File

@@ -3,6 +3,7 @@
import { getHordeModels } from "src/ts/horde/getModels"; import { getHordeModels } from "src/ts/horde/getModels";
import Arcodion from "./Arcodion.svelte"; import Arcodion from "./Arcodion.svelte";
import { language } from "src/lang"; import { language } from "src/lang";
import { isNodeServer, isTauri } from "src/ts/storage/globalApi";
export let value = "" export let value = ""
export let onChange: (v:string) => void = (v) => {} export let onChange: (v:string) => void = (v) => {}
@@ -24,6 +25,8 @@
return "Kobold" return "Kobold"
case "custom": case "custom":
return "Plugin" return "Plugin"
case "novelai":
return "NovelAI"
default: default:
if(name.startsWith("horde:::")){ if(name.startsWith("horde:::")){
return name.replace(":::", " ") return name.replace(":::", " ")
@@ -55,6 +58,9 @@
<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>
{#if isTauri ||isNodeServer}
<button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('novelai')}}>NovelAI Clio</button>
{/if}
<Arcodion name="Horde"> <Arcodion name="Horde">
{#await getHordeModels()} {#await getHordeModels()}
<button class="p-2">Loading...</button> <button class="p-2">Loading...</button>