diff --git a/invokeai/frontend/web/src/features/nodes/types/constants.ts b/invokeai/frontend/web/src/features/nodes/types/constants.ts index 9f6124c9d4..b864501803 100644 --- a/invokeai/frontend/web/src/features/nodes/types/constants.ts +++ b/invokeai/frontend/web/src/features/nodes/types/constants.ts @@ -17,6 +17,7 @@ export const FIELD_TYPE_MAP: Record = { ClipField: 'clip', VaeField: 'vae', model: 'model', + vae_model: 'vae_model', array: 'array', item: 'item', ColorField: 'color', @@ -116,6 +117,12 @@ export const FIELDS: Record = { 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'), diff --git a/invokeai/frontend/web/src/features/nodes/util/modelIdToVAEModelField.ts b/invokeai/frontend/web/src/features/nodes/util/modelIdToVAEModelField.ts index 3850ad443d..0cb608a936 100644 --- a/invokeai/frontend/web/src/features/nodes/util/modelIdToVAEModelField.ts +++ b/invokeai/frontend/web/src/features/nodes/util/modelIdToVAEModelField.ts @@ -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 diff --git a/invokeai/frontend/web/src/services/api/types.d.ts b/invokeai/frontend/web/src/services/api/types.d.ts index 3da844d764..18942a47d6 100644 --- a/invokeai/frontend/web/src/services/api/types.d.ts +++ b/invokeai/frontend/web/src/services/api/types.d.ts @@ -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