github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add Qwen3.6 VL captioner.

Browse Source
This commit is contained in:
Jaret Burkett
2026-07-15 07:02:33 -06:00
parent 18da85153b
commit 691ddf434e
5 changed files with 68 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
ui/src/components/CaptionSimpleJob.tsx
View File

@@ -183,6 +183,13 @@ const CaptionSimpleJob: React.FC<Props> = ({ jobConfig, setJobConfig, gpuIDs, se
checked={jobConfig.config.process[0].caption.compile || false}
onChange={value => setJobConfig(value, 'config.process[0].caption.compile')}
/>
{additionalSections.includes('caption.thinking') && (
<Checkbox
label="Thinking"
checked={jobConfig.config.process[0].caption.thinking || false}
onChange={value => setJobConfig(value, 'config.process[0].caption.thinking')}
/>
)}
</FormGroup>
</div>
</div>

6
ui/src/helpers/captionOptions.ts
View File

@@ -1,7 +1,7 @@
import { GroupedSelectOption, SelectOption } from "@/types";
type CaptionGroup = 'image' | 'music';
type AdditionalSections = 'caption.model_name_or_path2' | 'caption.caption_prompt' | 'caption.max_res' | 'caption.max_new_tokens' | 'caption.fixed_caption';
type AdditionalSections = 'caption.model_name_or_path2' | 'caption.caption_prompt' | 'caption.max_res' | 'caption.max_new_tokens' | 'caption.fixed_caption' | 'caption.thinking';
export interface CaptionOption {
name: string;
@@ -66,12 +66,16 @@ export const captionerTypes: CaptionOption[] = [
{ value: 'Qwen/Qwen3-VL-2B-Instruct', label: 'Qwen/Qwen3-VL-2B-Instruct' },
{ value: 'Qwen/Qwen3-VL-4B-Instruct', label: 'Qwen/Qwen3-VL-4B-Instruct' },
{ value: 'Qwen/Qwen3-VL-8B-Instruct', label: 'Qwen/Qwen3-VL-8B-Instruct' },
{ value: 'huihui-ai/Huihui-Qwen3-VL-8B-Instruct-abliterated', label: 'huihui-ai/Huihui-Qwen3-VL-8B-Instruct-abliterated' },
{ value: 'Qwen/Qwen3-VL-30B-A3B-Instruct', label: 'Qwen/Qwen3-VL-30B-A3B-Instruct' },
{ value: 'Qwen/Qwen3.6-27B', label: 'Qwen/Qwen3.6-27B' },
{ value: 'huihui-ai/Huihui-Qwen3.6-27B-abliterated', label: 'huihui-ai/Huihui-Qwen3.6-27B-abliterated' },
],
additionalSections: [
'caption.caption_prompt',
'caption.max_res',
'caption.max_new_tokens',
'caption.thinking',
],
},
{

1
ui/src/types.ts
View File

@@ -281,6 +281,7 @@ export interface CaptionProcessConfig {
max_new_tokens?: number;
fixed_caption?: string;
caption_extension?: string;
thinking?: boolean;
}
}