From 2db090144a8e6b568104ec5808a2f957545d9c50 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Tue, 27 Jan 2026 09:34:46 -0700
Subject: [PATCH] Add support for Z-Image

---
 ui/src/app/jobs/new/options.ts | 19 +++++++++++++++++++
 version.py                     |  2 +-
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/ui/src/app/jobs/new/options.ts b/ui/src/app/jobs/new/options.ts
index 64fa7cfa..c5738d19 100644
--- a/ui/src/app/jobs/new/options.ts
+++ b/ui/src/app/jobs/new/options.ts
@@ -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)',
diff --git a/version.py b/version.py
index ac8c2387..64e3bc05 100644
--- a/version.py
+++ b/version.py
@@ -1 +1 @@
-VERSION = "0.7.20"
+VERSION = "0.7.21"