Add model selection for VitsModel

This commit is contained in:
kwaroran
2024-01-06 06:45:18 +09:00
parent 66c6511684
commit 7344e566f4
7 changed files with 366 additions and 220 deletions

View File

@@ -29,6 +29,7 @@
import TriggerList from "./Scripts/TriggerList.svelte";
import CheckInput from "../UI/GUI/CheckInput.svelte";
import { updateInlayScreen } from "src/ts/process/inlayScreen";
import { registerOnnxModel } from "src/ts/process/embedding/transformers";
let subMenu = 0
@@ -626,6 +627,19 @@
<span class="text-textcolor">Language</span>
<TextInput additionalClass="mb-4 mt-2" bind:value={currentChar.data.hfTTS.language} placeholder="en" />
{/if}
{#if currentChar.data.ttsMode === 'vits'}
{#if currentChar.data.vits}
<span class="text-textcolor">{currentChar.data.vits.name ?? 'Unnamed VitsModel'}</span>
{:else}
<span class="text-textcolor">No Model</span>
{/if}
<Button on:click={async () => {
const model = await registerOnnxModel()
if(model && currentChar.type === 'character'){
currentChar.data.vits = model
}
}}>{language.selectModel}</Button>
{/if}
{#if currentChar.data.ttsMode}
<div class="flex items-center mt-2">
<Check bind:check={currentChar.data.ttsReadOnlyQuoted} name={language.ttsReadOnlyQuoted}/>