Add support for Baidu's ERNIE-Image (#793)

* Add support for ERNIE Image

* change float64 to float32

* Version bump

* Update ERNIE defaults
This commit is contained in:
Jaret Burkett (Ostris)
2026-04-14 09:45:12 -06:00
committed by GitHub
parent e868fca562
commit 3e0c904054
7 changed files with 834 additions and 4 deletions

View File

@@ -742,6 +742,28 @@ export const modelArchs: ModelArch[] = [
'model.qie.match_target_res',
],
},
{
name: 'ernie_image',
label: 'ERNIE-Image',
group: 'image',
defaults: {
// default updates when [selected, unselected] in the UI
'config.process[0].model.name_or_path': ['baidu/ERNIE-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: 'flux2_klein_9b',
label: 'FLUX.2-klein-base-9B',