Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Missing VAE Loader stuff

Browse Source
This commit is contained in:
blessedcoolant 2023-07-01 10:06:18 +12:00 committed by psychedelicious
parent 6c6299ce49
commit 630f3c8b0b
3 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/frontend/web/src/features/nodes/types/constants.ts
View File

@ -17,6 +17,7 @@ export const FIELD_TYPE_MAP: Record<string, FieldType> = {
ClipField: 'clip',
VaeField: 'vae',
model: 'model',
vae_model: 'vae_model',
array: 'array',
item: 'item',
ColorField: 'color',
@ -116,6 +117,12 @@ export const FIELDS: Record<FieldType, FieldUIConfig> = {
title: 'Model',
description: 'Models are models.',
},
vae_model: {
color: 'teal',
colorCssVar: getColorTokenCssVariable('teal'),
title: 'Model',
description: 'Models are models.',
},
array: {
color: 'gray',
colorCssVar: getColorTokenCssVariable('gray'),

2
invokeai/frontend/web/src/features/nodes/util/modelIdToVAEModelField.ts
View File

@ -1,4 +1,4 @@
import { BaseModelType } from 'services/api/types';
import { BaseModelType, VAEModelField } from 'services/api/types';
/**
* Crudely converts a model id to a main model field

1
invokeai/frontend/web/src/services/api/types.d.ts vendored
View File

@ -34,6 +34,7 @@ export type OffsetPaginatedResults_ImageDTO_ =
export type ModelType = S<'ModelType'>;
export type BaseModelType = S<'BaseModelType'>;
export type MainModelField = S<'MainModelField'>;
export type VAEModelField = S<'VAEModelField'>;
export type ModelsList = S<'ModelsList'>;
// Graphs