fix the ctrl tuff maybe

ui/src/app/jobs/new/SimplifiedJob.tsx

@@ -22,7 +22,16 @@ const ACCEPTED_FILE_TYPES = {
   'text/*': ['.txt'],
 };
 
-const FOUR_BATCH_ARCHES = new Set(['flux', 'flux_kontext', 'flex1', 'flex2', 'wan21:1b']);
+const FOUR_BATCH_ARCHES = new Set([
+  'flux',
+  'flux_kontext',
+  'flex1',
+  'flex2',
+  'wan21:1b',
+  'qwen_image',
+  'qwen_image_edit',
+  'qwen_image_edit_plus',
+]);
 
 const resolveBatchSettings = (arch: string) =>
   FOUR_BATCH_ARCHES.has(arch)
@@ -256,16 +265,6 @@ export default function SimplifiedJob({
     }
   }, [jobConfig.config.name, trainDatasetNameTouched, controlDatasetNameTouched]);
 
-  useEffect(() => {
-    const { batchSize, learningRate } = resolveBatchSettings(process.model.arch);
-    if (process.train.batch_size !== batchSize) {
-      setJobConfig(batchSize, 'config.process[0].train.batch_size');
-    }
-    if (process.train.lr !== learningRate) {
-      setJobConfig(learningRate, 'config.process[0].train.lr');
-    }
-  }, [process.model.arch, process.train.batch_size, process.train.lr, setJobConfig]);
-
   const ensureDataset = useCallback(
     async (
       datasetName: string,

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

@@ -53,7 +53,7 @@ export const defaultJobConfig: JobConfig = {
         },
         datasets: [defaultDatasetConfig],
         train: {
-          batch_size: 1,
+          batch_size: 4,
           bypass_guidance_embedding: true,
           steps: 1200,
           gradient_accumulation: 1,
@@ -69,7 +69,7 @@ export const defaultJobConfig: JobConfig = {
           },
           unload_text_encoder: false,
           cache_text_embeddings: false,
-          lr: 0.0001,
+          lr: 0.0003,
           ema_config: {
             use_ema: false,
             ema_decay: 0.99,

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

@@ -369,7 +369,7 @@ export const modelArchs: ModelArch[] = [
       'config.process[0].train.lr': [0.0003, 0.0001],
     },
     disableSections: ['network.conv', 'train.unload_text_encoder'],
-    additionalSections: ['datasets.multi_control_paths', 'sample.multi_ctrl_imgs', 'model.low_vram'],
+    additionalSections: ['datasets.control_path', 'sample.ctrl_img', 'model.low_vram'],
     accuracyRecoveryAdapters: {
       '3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/qwen_image_edit_2509_torchao_uint3.safetensors',
     },