ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for training Qwen Image Edit in the UI

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-22 10:26:46 -06:00
parent aa99784b89
commit 59ff4efae5
2 changed files with 25 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
ui/src/app/jobs/new/options.ts
View File

@@ -10,7 +10,7 @@ type AdditionalSections =
| 'datasets.num_frames'
| 'model.multistage'
| 'model.low_vram';
type ModelGroup = 'image' | 'video';
type ModelGroup = 'image' | 'instruction' | 'video';
export interface ModelArch {
name: string;
@@ -44,7 +44,7 @@ export const modelArchs: ModelArch[] = [
{
name: 'flux_kontext',
label: 'FLUX.1-Kontext-dev',
group: 'image',
group: 'instruction',
defaults: {
// default updates when [selected, unselected] in the UI
'config.process[0].model.name_or_path': ['black-forest-labs/FLUX.1-Kontext-dev', defaultNameOrPath],
@@ -306,6 +306,27 @@ export const modelArchs: ModelArch[] = [
'3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_torchao_uint3.safetensors',
},
},
{
name: 'qwen_image_edit',
label: 'Qwen-Image-Edit',
group: 'instruction',
defaults: {
// default updates when [selected, unselected] in the UI
'config.process[0].model.name_or_path': ['Qwen/Qwen-Image-Edit', defaultNameOrPath],
'config.process[0].model.quantize': [true, false],
'config.process[0].model.quantize_te': [true, false],
'config.process[0].model.low_vram': [true, false],
'config.process[0].sample.sampler': ['flowmatch', 'flowmatch'],
'config.process[0].train.noise_scheduler': ['flowmatch', 'flowmatch'],
'config.process[0].train.timestep_type': ['weighted', 'sigmoid'],
'config.process[0].model.qtype': ['qfloat8', 'qfloat8'],
},
disableSections: ['network.conv'],
additionalSections: ['datasets.control_path', 'sample.ctrl_img', 'model.low_vram'],
accuracyRecoveryAdapters: {
'3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_edit_torchao_uint3.safetensors',
},
},
{
name: 'hidream',
label: 'HiDream',
@@ -327,7 +348,7 @@ export const modelArchs: ModelArch[] = [
{
name: 'hidream_e1',
label: 'HiDream E1',
group: 'image',
group: 'instruction',
defaults: {
// default updates when [selected, unselected] in the UI
'config.process[0].model.name_or_path': ['HiDream-ai/HiDream-E1-1', defaultNameOrPath],

2
version.py
View File

@@ -1 +1 @@
VERSION = "0.5.3"
VERSION = "0.5.4"