diff --git a/invokeai/frontend/web/src/features/nodes/components/InputFieldComponent.tsx b/invokeai/frontend/web/src/features/nodes/components/InputFieldComponent.tsx index 44c5335185..b179adff23 100644 --- a/invokeai/frontend/web/src/features/nodes/components/InputFieldComponent.tsx +++ b/invokeai/frontend/web/src/features/nodes/components/InputFieldComponent.tsx @@ -17,6 +17,7 @@ import ModelInputFieldComponent from './fields/ModelInputFieldComponent'; import NumberInputFieldComponent from './fields/NumberInputFieldComponent'; import StringInputFieldComponent from './fields/StringInputFieldComponent'; import UnetInputFieldComponent from './fields/UnetInputFieldComponent'; +import VaeInputFieldComponent from './fields/VaeInputFieldComponent'; import VaeModelInputFieldComponent from './fields/VaeModelInputFieldComponent'; type InputFieldComponentProps = { @@ -123,9 +124,9 @@ const InputFieldComponent = (props: InputFieldComponentProps) => { ); } - if (type === 'vae_model' && template.type === 'vae_model') { + if (type === 'vae' && template.type === 'vae') { return ( - +) => { + const { nodeId, field } = props; + return null; +}; + +export default memo(VaeInputFieldComponent); diff --git a/invokeai/frontend/web/src/features/nodes/types/types.ts b/invokeai/frontend/web/src/features/nodes/types/types.ts index dda7bff527..4c47c63068 100644 --- a/invokeai/frontend/web/src/features/nodes/types/types.ts +++ b/invokeai/frontend/web/src/features/nodes/types/types.ts @@ -94,7 +94,7 @@ export type InputFieldValue = | ConditioningInputFieldValue | UNetInputFieldValue | ClipInputFieldValue - | VaeModelInputFieldValue + | VaeInputFieldValue | ControlInputFieldValue | EnumInputFieldValue | MainModelInputFieldValue @@ -121,7 +121,7 @@ export type InputFieldTemplate = | ConditioningInputFieldTemplate | UNetInputFieldTemplate | ClipInputFieldTemplate - | VaeModelInputFieldTemplate + | VaeInputFieldTemplate | ControlInputFieldTemplate | EnumInputFieldTemplate | ModelInputFieldTemplate