diff --git a/src/lib/types/meta.ts b/src/lib/types/meta.ts index 83b9489..304cb5a 100644 --- a/src/lib/types/meta.ts +++ b/src/lib/types/meta.ts @@ -1,6 +1,8 @@ +import type { TargetModels } from "$lib/types/models"; + export type DatasetMeta = { name: string; - targetModel: string; + targetModel: TargetModels; loraType: string; unifiedImageSize: boolean; unifiedImageRatio: boolean; diff --git a/src/routes/create/+page.svelte b/src/routes/create/+page.svelte index 799f4f9..fadc545 100644 --- a/src/routes/create/+page.svelte +++ b/src/routes/create/+page.svelte @@ -4,6 +4,7 @@ import Close from '$lib/components/icons/Close.svelte'; import { activeDatasetMeta, openedDatasetDir, updateActiveDatasetMeta } from '$lib/stores/dataset'; import { currentActivate } from '$lib/stores/navigate'; import type { DatasetMeta } from '$lib/types/meta'; +import { ModelChoices } from '$lib/types/models'; import { message, open } from '@tauri-apps/plugin-dialog'; import { readDir } from '@tauri-apps/plugin-fs'; import { isNil } from 'es-toolkit'; @@ -14,6 +15,7 @@ afterNavigate(() => { let storePath = $state(''); let datasetName = $state(''); +let targetModel = $state(ModelChoices[0].value); async function backToBoot() { await goto('/boot'); @@ -58,7 +60,7 @@ async function createDataset() { if (!isNil(storePath) && !isNil(datasetName) && storePath.trim() && datasetName.trim()) { const datasetMeta: DatasetMeta = { name: datasetName.trim(), - targetModel: '', + targetModel: targetModel, loraType: '', unifiedImageSize: true, unifiedImageRatio: true, @@ -109,5 +111,13 @@ async function createDataset() { Dataset Name +