ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-02-23 05:43:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add vram flag to some models in the ui

Browse Source
This commit is contained in:
Jaret Burkett
2025-07-24 07:02:46 -06:00
parent c5eb763342
commit 523c159579
3 changed files with 23 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
ui/src/app/jobs/new/SimpleJob.tsx
View File

@@ -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

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

@@ -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',

2
version.py
View File

@@ -1 +1 @@
VERSION = "0.3.9"
VERSION = "0.3.10"