Add support for training directly on Krea2 Turbo with a training adapter

ui/src/app/jobs/new/options.ts

@@ -1041,7 +1041,7 @@ export const modelArchs: ModelArch[] = [
   },
   {
     name: 'krea2',
-    label: 'Krea 2 (K2)',
+    label: 'Krea 2 (raw)',
     group: 'image',
     defaults: {
       'config.process[0].model.name_or_path': ['krea/Krea-2-Raw', defaultNameOrPath],
@@ -1060,6 +1060,34 @@ export const modelArchs: ModelArch[] = [
       'model.layer_offloading',
     ],
   },
+  {
+    name: 'krea2:turbo',
+    label: 'Krea 2 Turbo (w/ Training Adapter)',
+    group: 'image',
+    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],
+    },
+    disableSections: [
+      'network.conv',
+    ],
+    additionalSections: [
+      'model.low_vram',
+      'model.layer_offloading',
+      'model.assistant_lora_path'
+    ],
+  },
   {
     name: 'boogu_image',
     label: 'Boogu Image',