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:
committed by
GitHub
parent
e868fca562
commit
3e0c904054
@@ -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',
|
||||
|
||||
Reference in New Issue
Block a user