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
+