github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add support for Mage-Flow and Mage-Flow Edit

Browse Source
This commit is contained in:
Jaret Burkett
2026-07-25 11:34:20 -06:00
parent 0c3a5e6970
commit 92bdb6e473
12 changed files with 2690 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
ui/src/app/jobs/new/options.ts
View File

@@ -1196,6 +1196,55 @@ export const modelArchs: ModelArch[] = [
'model.model_kwargs.kv_cache',
],
},
{
name: 'mageflow',
label: 'Mage-Flow',
group: 'image',
defaults: {
'config.process[0].model.name_or_path': ['microsoft/Mage-Flow-Base', defaultNameOrPath],
'config.process[0].model.quantize': [true, false],
'config.process[0].model.quantize_te': [true, false],
'config.process[0].train.timestep_type': ['linear', 'sigmoid'],
'config.process[0].network.conv': [undefined, 16],
'config.process[0].network.conv_alpha': [undefined, 16],
'config.process[0].model.low_vram': [true, false],
'config.process[0].sample.guidance_scale': [4, 4],
'config.process[0].sample.sample_steps': [25, 25],
},
disableSections: [
'network.conv',
],
additionalSections: [
'model.low_vram',
'model.layer_offloading',
],
},
{
name: 'mageflow_edit',
label: 'Mage-Flow Edit',
group: 'instruction',
defaults: {
'config.process[0].model.name_or_path': ['microsoft/Mage-Flow-Edit-Base', defaultNameOrPath],
'config.process[0].model.quantize': [true, false],
'config.process[0].model.quantize_te': [true, false],
'config.process[0].train.timestep_type': ['linear', 'sigmoid'],
'config.process[0].network.conv': [undefined, 16],
'config.process[0].network.conv_alpha': [undefined, 16],
'config.process[0].model.low_vram': [true, false],
'config.process[0].sample.guidance_scale': [4, 4],
'config.process[0].sample.sample_steps': [25, 25],
'config.process[0].train.unload_text_encoder': [false, false],
},
disableSections: [
'network.conv', 'train.unload_text_encoder',
],
additionalSections: [
'datasets.multi_control_paths',
'sample.multi_ctrl_imgs',
'model.low_vram',
'model.layer_offloading',
],
},
{
name: 'boogu_image',
label: 'Boogu Image',