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

Add support for WebLLM

Browse Source
This commit is contained in:
kwaroran
2024-01-09 00:27:58 +09:00
parent 1ee7107153
commit 350bc66851
4 changed files with 32 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,7 +3,6 @@
import { getHordeModels } from "src/ts/horde/getModels";
import Arcodion from "./Arcodion.svelte";
import { language } from "src/lang";
import { isNodeServer, isTauri } from "src/ts/storage/globalApi";
import Help from "../Others/Help.svelte";
import CheckInput from "./GUI/CheckInput.svelte";
@@ -86,6 +85,10 @@
const split = name.split(":::")
return `Horde ${split[1]}`
}
if(name.startsWith('tf:::')){
const split = name.split(":::")
return `${split[1]}`
}
return name
}
}
@@ -190,7 +193,14 @@
</button>
{/each}
{/await}
</Arcodion>
</Arcodion>
{#if showUnrec}
<Arcodion name="WebLLM Local">
<button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('hf:::Xenova/opt-350m')}}>opt-350m</button>
<button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('hf:::Xenova/tiny-random-mistral')}}>tiny-random-mistral</button>
<button class="hover:bg-selected px-6 py-2 text-lg" on:click={() => {changeModel('hf:::Xenova/gpt2-large-conversational')}}>gpt2-large-conversational</button>
</Arcodion>
{/if}
{#if $DataBase.plugins.length > 0}
<button on:click={() => {changeModel('custom')}} class="hover:bg-selected px-6 py-2 text-lg" >Plugin</button>
{/if}