Add support in UI for Hidream E1

This commit is contained in:
Jaret Burkett
2025-07-27 18:13:36 -06:00
parent ed8d14225f
commit 99705ec8be
2 changed files with 19 additions and 1 deletions

View File

@@ -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',

View File

@@ -1 +1 @@
VERSION = "0.3.12"
VERSION = "0.3.13"