From 523c1595794e91b7cc357193e99e5d2fcb77c5af Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Thu, 24 Jul 2025 07:02:46 -0600
Subject: [PATCH] Add vram flag to some models in the ui

---
 ui/src/app/jobs/new/SimpleJob.tsx | 19 +++++++++++++++++--
 ui/src/app/jobs/new/options.ts    | 10 +++++-----
 version.py                        |  2 +-
 3 files changed, 23 insertions(+), 8 deletions(-)

diff --git a/ui/src/app/jobs/new/SimpleJob.tsx b/ui/src/app/jobs/new/SimpleJob.tsx
index 52f6454a..9d192f67 100644
--- a/ui/src/app/jobs/new/SimpleJob.tsx
+++ b/ui/src/app/jobs/new/SimpleJob.tsx
@@ -86,13 +86,19 @@ export default function SimpleJob({
                 if (!currentArch || currentArch.name === value) {
                   return;
                 }
+                // update the defaults when a model is selected
+                const newArch = modelArchs.find(model => model.name === value);
+
+                 // update vram setting
+                if (!(newArch?.additionalSections?.includes('model.low_vram'))) {
+                  setJobConfig(false, 'config.process[0].model.low_vram');
+                }
 
                 // revert defaults from previous model
                 for (const key in currentArch.defaults) {
                   setJobConfig(currentArch.defaults[key][1], key);
                 }
-                // update the defaults when a model is selected
-                const newArch = modelArchs.find(model => model.name === value);
+
                 if (newArch?.defaults) {
                   for (const key in newArch.defaults) {
                     setJobConfig(newArch.defaults[key][0], key);
@@ -160,6 +166,15 @@ export default function SimpleJob({
                 </div>
               </FormGroup>
             )}
+            {modelArch?.additionalSections?.includes('model.low_vram') && (
+              <FormGroup label="Options">
+                  <Checkbox
+                    label="Low VRAM"
+                    checked={jobConfig.config.process[0].model.low_vram}
+                    onChange={value => setJobConfig(value, 'config.process[0].model.low_vram')}
+                  />
+              </FormGroup>
+            )}
           </Card>
           <Card title="Target Configuration">
             <SelectInput
diff --git a/ui/src/app/jobs/new/options.ts b/ui/src/app/jobs/new/options.ts
index dbfe311b..93f5f713 100644
--- a/ui/src/app/jobs/new/options.ts
+++ b/ui/src/app/jobs/new/options.ts
@@ -3,7 +3,7 @@ import { GroupedSelectOption } from "@/types";
 type Control = 'depth' | 'line' | 'pose' | 'inpaint';
 
 type DisableableSections = 'model.quantize' | 'train.timestep_type' | 'network.conv';
-type AdditionalSections = 'datasets.control_path' | 'sample.ctrl_img' | 'datasets.num_frames';
+type AdditionalSections = 'datasets.control_path' | 'sample.ctrl_img' | 'datasets.num_frames' | 'model.low_vram';
 type ModelGroup = 'image' | 'video';
 
 export interface ModelArch {
@@ -126,7 +126,7 @@ export const modelArchs: ModelArch[] = [
       'config.process[0].sample.fps': [15, 1],
     },
     disableSections: ['network.conv'],
-    additionalSections: ['datasets.num_frames'],
+    additionalSections: ['datasets.num_frames', 'model.low_vram'],
   },
   {
     name: 'wan21_i2v:14b480p',
@@ -145,7 +145,7 @@ export const modelArchs: ModelArch[] = [
       'config.process[0].train.timestep_type': ['weighted', 'sigmoid'],
     },
     disableSections: ['network.conv'],
-    additionalSections: ['sample.ctrl_img', 'datasets.num_frames'],
+    additionalSections: ['sample.ctrl_img', 'datasets.num_frames', 'model.low_vram'],
   },
   {
     name: 'wan21_i2v:14b',
@@ -164,7 +164,7 @@ export const modelArchs: ModelArch[] = [
       'config.process[0].train.timestep_type': ['weighted', 'sigmoid'],
     },
     disableSections: ['network.conv'],
-    additionalSections: ['sample.ctrl_img', 'datasets.num_frames'],
+    additionalSections: ['sample.ctrl_img', 'datasets.num_frames', 'model.low_vram'],
   },
   {
     name: 'wan21:14b',
@@ -182,7 +182,7 @@ export const modelArchs: ModelArch[] = [
       'config.process[0].sample.fps': [15, 1],
     },
     disableSections: ['network.conv'],
-    additionalSections: ['datasets.num_frames'],
+    additionalSections: ['datasets.num_frames', 'model.low_vram'],
   },
   {
     name: 'lumina2',
diff --git a/version.py b/version.py
index 9c170088..14adb8a1 100644
--- a/version.py
+++ b/version.py
@@ -1 +1 @@
-VERSION = "0.3.9"
\ No newline at end of file
+VERSION = "0.3.10"
\ No newline at end of file