Add support for training lodestones/Zeta-Chroma

ui/src/app/jobs/new/options.ts

@@ -29,7 +29,8 @@ type AdditionalSections =
   | 'model.low_vram'
   | 'model.qie.match_target_res'
   | 'model.assistant_lora_path';
-type ModelGroup = 'image' | 'instruction' | 'video';
+
+type ModelGroup = 'image' | 'instruction' | 'video' | 'experimental';
 
 export interface ModelArch {
   name: string;
@@ -133,6 +134,21 @@ export const modelArchs: ModelArch[] = [
     },
     disableSections: ['network.conv'],
   },
+  {
+    name: 'zeta_chroma',
+    label: 'Zeta Chroma',
+    group: 'experimental',
+    defaults: {
+      // default updates when [selected, unselected] in the UI
+      'config.process[0].model.name_or_path': ['lodestones/Zeta-Chroma/zeta-chroma-base-x0-pixel-dino-distance.safetensors', defaultNameOrPath],
+      'config.process[0].model.extras_name_or_path': ['Tongyi-MAI/Z-Image-Turbo', undefined],
+      'config.process[0].model.quantize': [true, false],
+      'config.process[0].model.quantize_te': [true, false],
+      'config.process[0].sample.sampler': ['flowmatch', 'flowmatch'],
+      'config.process[0].train.noise_scheduler': ['flowmatch', 'flowmatch'],
+    },
+    disableSections: ['network.conv'],
+  },
   {
     name: 'wan21:1b',
     label: 'Wan 2.1 (1.3B)',