Add model size selection to GenerationForm and enhance model management features

This commit is contained in:
Jamie Pine
2026-01-25 03:19:27 -08:00
parent 6429cb6673
commit 6164877f7f
5 changed files with 65 additions and 18 deletions
@@ -31,6 +31,7 @@ const generationSchema = z.object({
text: z.string().min(1, 'Text is required').max(5000),
language: z.enum(['en', 'zh']),
seed: z.number().int().optional(),
modelSize: z.enum(['1.7B', '0.6B']).optional(),
});
type GenerationFormValues = z.infer<typeof generationSchema>;
@@ -47,6 +48,7 @@ export function GenerationForm() {
text: '',
language: 'en',
seed: undefined,
modelSize: '1.7B',
},
});
@@ -57,6 +59,7 @@ export function GenerationForm() {
text: data.text,
language: data.language,
seed: data.seed,
model_size: data.modelSize,
});
toast({
@@ -126,7 +129,7 @@ export function GenerationForm() {
)}
/>
<div className="grid gap-4 md:grid-cols-2">
<div className="grid gap-4 md:grid-cols-3">
<FormField
control={form.control}
name="language"
@@ -149,6 +152,29 @@ export function GenerationForm() {
)}
/>
<FormField
control={form.control}
name="modelSize"
render={({ field }) => (
<FormItem>
<FormLabel>Model Size</FormLabel>
<Select onValueChange={field.onChange} defaultValue={field.value}>
<FormControl>
<SelectTrigger>
<SelectValue />
</SelectTrigger>
</FormControl>
<SelectContent>
<SelectItem value="1.7B">Qwen TTS 1.7B (Higher Quality)</SelectItem>
<SelectItem value="0.6B">Qwen TTS 0.6B (Faster)</SelectItem>
</SelectContent>
</Select>
<FormDescription>Larger models produce better quality</FormDescription>
<FormMessage />
</FormItem>
)}
/>
<FormField
control={form.control}
name="seed"