From 99705ec8be84a000b7ce58cad80436c1b88425ce Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Sun, 27 Jul 2025 18:13:36 -0600
Subject: [PATCH] Add support in UI for Hidream E1

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

diff --git a/ui/src/app/jobs/new/options.ts b/ui/src/app/jobs/new/options.ts
index e9d00220..c52aa523 100644
--- a/ui/src/app/jobs/new/options.ts
+++ b/ui/src/app/jobs/new/options.ts
@@ -212,6 +212,24 @@ export const modelArchs: ModelArch[] = [
     },
     disableSections: ['network.conv'],
   },
+  {
+    name: 'hidream_e1',
+    label: 'HiDream E1',
+    group: 'image',
+    defaults: {
+      // default updates when [selected, unselected] in the UI
+      'config.process[0].model.name_or_path': ['HiDream-ai/HiDream-E1-1', defaultNameOrPath],
+      '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'],
+      'config.process[0].train.lr': [0.0001, 0.0001],
+      'config.process[0].train.timestep_type': ['weighted', 'sigmoid'],
+      'config.process[0].network.network_kwargs.ignore_if_contains': [['ff_i.experts', 'ff_i.gate'], []],
+    },
+    disableSections: ['network.conv'],
+    additionalSections: ['datasets.control_path', 'sample.ctrl_img'],
+  },
   {
     name: 'sdxl',
     label: 'SDXL',
diff --git a/version.py b/version.py
index 7a076d2f..b654c117 100644
--- a/version.py
+++ b/version.py
@@ -1 +1 @@
-VERSION = "0.3.12"
\ No newline at end of file
+VERSION = "0.3.13"
\ No newline at end of file