From 3a94591c89b05796bb000dc09f81cd0acdfc1cda Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Sat, 4 Jul 2026 09:17:15 -0600
Subject: [PATCH] Disable text encoder unload with krea2 o-edit

---
 ui/src/app/jobs/new/options.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/ui/src/app/jobs/new/options.ts b/ui/src/app/jobs/new/options.ts
index c28ae30..7f52db0 100644
--- a/ui/src/app/jobs/new/options.ts
+++ b/ui/src/app/jobs/new/options.ts
@@ -1100,6 +1100,7 @@ export const modelArchs: ModelArch[] = [
       '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].train.unload_text_encoder': [false, false],
       'config.process[0].model.model_kwargs': [
         {
           edit: true,
@@ -1109,7 +1110,7 @@ export const modelArchs: ModelArch[] = [
       ],
     },
     disableSections: [
-      'network.conv',
+      'network.conv', 'train.unload_text_encoder'
     ],
     additionalSections: [
       'datasets.multi_control_paths',
@@ -1137,6 +1138,7 @@ export const modelArchs: ModelArch[] = [
       ],
       'config.process[0].sample.guidance_scale': [1, 4],
       'config.process[0].sample.sample_steps': [8, 25],
+      'config.process[0].train.unload_text_encoder': [false, false],
       'config.process[0].model.model_kwargs': [
         {
           edit: true,
@@ -1146,7 +1148,7 @@ export const modelArchs: ModelArch[] = [
       ],
     },
     disableSections: [
-      'network.conv',
+      'network.conv', 'train.unload_text_encoder'
     ],
     additionalSections: [
       'datasets.multi_control_paths',