From c8576425a45f29c47619694ae56a1c0238c9a682 Mon Sep 17 00:00:00 2001
From: Vixalie <vixalie@outlook.com>
Date: Sat, 28 Mar 2026 08:30:43 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=9B=B4=E6=96=B0=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E9=9B=86=E5=88=9B=E5=BB=BA=E5=8A=9F=E8=83=BD=EF=BC=8C=E4=BD=BF?=
 =?UTF-8?q?=E7=94=A8=20TargetModels=20=E7=B1=BB=E5=9E=8B=E6=9B=BF=E4=BB=A3?=
 =?UTF-8?q?=E5=AD=97=E7=AC=A6=E4=B8=B2=E4=BD=9C=E4=B8=BA=E7=9B=AE=E6=A0=87?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/lib/types/meta.ts          |  4 +++-
 src/routes/create/+page.svelte | 12 +++++++++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

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() {
     <span class="label min-w-[10em]">Dataset Name</span>
     <input type="text" class="min-w-[20em]" bind:value={datasetName} />
   </label>
+  <label class="input input-md focus-within:outline-0 w-fit">
+    <span class="label min-w-[10em]">Target Model</span>
+    <select class="min-w-[20em] focus:outline-0" bind:value={targetModel}>
+      {#each ModelChoices as model}
+        <option value={model.value}>{model.label}</option>
+      {/each}
+    </select>
+  </label>
   <button class="btn btn-primary btn-md w-fit self-center" onclick={createDataset}>Create</button>
 </main>