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

Added support to train Krea2 as an edit model

Browse Source
This commit is contained in:
Jaret Burkett
2026-07-04 09:12:39 -06:00
parent f63221e577
commit b1e1a834d4
7 changed files with 519 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1085,7 +1085,76 @@ export const modelArchs: ModelArch[] = [
additionalSections: [
'model.low_vram',
'model.layer_offloading',
'model.assistant_lora_path'
'model.assistant_lora_path',
],
},
{
name: 'krea2:o_edit',
label: 'Krea 2 (raw) [Edit Training]',
group: 'experimental',
defaults: {
'config.process[0].model.name_or_path': ['krea/Krea-2-Raw', 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].model.model_kwargs': [
{
edit: true,
match_target_res: false,
},
{},
],
},
disableSections: [
'network.conv',
],
additionalSections: [
'datasets.multi_control_paths',
'sample.multi_ctrl_imgs',
'model.low_vram',
'model.layer_offloading',
'model.qie.match_target_res',
],
},
{
name: 'krea2:o_edit_turbo',
label: 'Krea 2 Turbo (w/ Training Adapter) [Edit Training]',
group: 'experimental',
defaults: {
'config.process[0].model.name_or_path': ['krea/Krea-2-Turbo', 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].model.assistant_lora_path': [
'ostris/krea2_turbo_training_adapter/krea2_turbo_training_adapter_v1.safetensors',
undefined,
],
'config.process[0].sample.guidance_scale': [1, 4],
'config.process[0].sample.sample_steps': [8, 25],
'config.process[0].model.model_kwargs': [
{
edit: true,
match_target_res: false,
},
{},
],
},
disableSections: [
'network.conv',
],
additionalSections: [
'datasets.multi_control_paths',
'sample.multi_ctrl_imgs',
'model.low_vram',
'model.layer_offloading',
'model.assistant_lora_path',
'model.qie.match_target_res',
],
},
{