ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-02-26 23:33:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for Z-Image

Browse Source
This commit is contained in:
Jaret Burkett
2026-01-27 09:34:46 -07:00
parent 9ef6f1a828
commit 2db090144a
2 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -583,6 +583,25 @@ export const modelArchs: ModelArch[] = [
disableSections: ['network.conv'],
additionalSections: ['model.low_vram', 'model.layer_offloading', 'model.assistant_lora_path'],
},
{
name: 'zimage',
label: 'Z-Image',
group: 'image',
defaults: {
// default updates when [selected, unselected] in the UI
'config.process[0].model.name_or_path': ['Tongyi-MAI/Z-Image', 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].train.unload_text_encoder': [false, 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: ['model.low_vram', 'model.layer_offloading'],
},
{
name: 'zimage:deturbo',
label: 'Z-Image De-Turbo (De-Distilled)',

2
version.py
View File

@@ -1 +1 @@
VERSION = "0.7.20"
VERSION = "0.7.21"