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: Type errors & missing Unet field component

Browse Source
This commit is contained in:
blessedcoolant 2023-07-14 18:00:41 +12:00 committed by psychedelicious
parent 8dd4ca5723
commit e1d6c09ed2
4 changed files with 25 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
invokeai/frontend/web/src/features/nodes/components/InputFieldComponent.tsx
View File

@ -16,8 +16,7 @@ import LoRAModelInputFieldComponent from './fields/LoRAModelInputFieldComponent'
import ModelInputFieldComponent from './fields/ModelInputFieldComponent'; import ModelInputFieldComponent from './fields/ModelInputFieldComponent';
import NumberInputFieldComponent from './fields/NumberInputFieldComponent'; import NumberInputFieldComponent from './fields/NumberInputFieldComponent';
import StringInputFieldComponent from './fields/StringInputFieldComponent'; import StringInputFieldComponent from './fields/StringInputFieldComponent';
import UNetInputFieldComponent from './fields/UNetInputFieldComponent'; import UnetInputFieldComponent from './fields/UnetInputFieldComponent';
import VaeInputFieldComponent from './fields/VaeInputFieldComponent';
import VaeModelInputFieldComponent from './fields/VaeModelInputFieldComponent'; import VaeModelInputFieldComponent from './fields/VaeModelInputFieldComponent';
type InputFieldComponentProps = { type InputFieldComponentProps = {
@ -106,7 +105,7 @@ const InputFieldComponent = (props: InputFieldComponentProps) => {
if (type === 'unet' && template.type === 'unet') { if (type === 'unet' && template.type === 'unet') {
return ( return (
<UNetInputFieldComponent <UnetInputFieldComponent
nodeId={nodeId} nodeId={nodeId}
field={field} field={field}
template={template} template={template}
@ -124,9 +123,9 @@ const InputFieldComponent = (props: InputFieldComponentProps) => {
); );
} }
if (type === 'vae' && template.type === 'vae') { if (type === 'vae_model' && template.type === 'vae_model') {
return ( return (
<VaeInputFieldComponent <VaeModelInputFieldComponent
nodeId={nodeId} nodeId={nodeId}
field={field} field={field}
template={template} template={template}

8
invokeai/frontend/web/src/features/nodes/components/fields/LoRAModelInputFieldComponent.tsx
View File

@ -5,8 +5,8 @@ import IAIMantineSelect from 'common/components/IAIMantineSelect';
import IAIMantineSelectItemWithTooltip from 'common/components/IAIMantineSelectItemWithTooltip'; import IAIMantineSelectItemWithTooltip from 'common/components/IAIMantineSelectItemWithTooltip';
import { fieldValueChanged } from 'features/nodes/store/nodesSlice'; import { fieldValueChanged } from 'features/nodes/store/nodesSlice';
import { import {
VaeModelInputFieldTemplate, LoRAModelInputFieldTemplate,
VaeModelInputFieldValue, LoRAModelInputFieldValue,
} from 'features/nodes/types/types'; } from 'features/nodes/types/types';
import { MODEL_TYPE_MAP } from 'features/parameters/types/constants'; import { MODEL_TYPE_MAP } from 'features/parameters/types/constants';
import { modelIdToLoRAModelParam } from 'features/parameters/util/modelIdToLoRAModelParam'; import { modelIdToLoRAModelParam } from 'features/parameters/util/modelIdToLoRAModelParam';
@ -17,8 +17,8 @@ import { FieldComponentProps } from './types';
const LoRAModelInputFieldComponent = ( const LoRAModelInputFieldComponent = (
props: FieldComponentProps< props: FieldComponentProps<
VaeModelInputFieldValue, LoRAModelInputFieldValue,
VaeModelInputFieldTemplate LoRAModelInputFieldTemplate
> >
) => { ) => {
const { nodeId, field } = props; const { nodeId, field } = props;

15
invokeai/frontend/web/src/features/nodes/components/fields/UnetInputFieldComponent.tsx Normal file
View File

@ -0,0 +1,15 @@
import {
UNetInputFieldTemplate,
UNetInputFieldValue,
} from 'features/nodes/types/types';
import { memo } from 'react';
import { FieldComponentProps } from './types';
const UNetInputFieldComponent = (
props: FieldComponentProps<UNetInputFieldValue, UNetInputFieldTemplate>
) => {
const { nodeId, field } = props;
return null;
};
export default memo(UNetInputFieldComponent);

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

@ -94,7 +94,7 @@ export type InputFieldValue =
| ConditioningInputFieldValue | ConditioningInputFieldValue
| UNetInputFieldValue | UNetInputFieldValue
| ClipInputFieldValue | ClipInputFieldValue
| VaeInputFieldValue | VaeModelInputFieldValue
| ControlInputFieldValue | ControlInputFieldValue
| EnumInputFieldValue | EnumInputFieldValue
| MainModelInputFieldValue | MainModelInputFieldValue
@ -121,7 +121,7 @@ export type InputFieldTemplate =
| ConditioningInputFieldTemplate | ConditioningInputFieldTemplate
| UNetInputFieldTemplate | UNetInputFieldTemplate
| ClipInputFieldTemplate | ClipInputFieldTemplate
| VaeInputFieldTemplate | VaeModelInputFieldTemplate
| ControlInputFieldTemplate | ControlInputFieldTemplate
| EnumInputFieldTemplate | EnumInputFieldTemplate
| ModelInputFieldTemplate | ModelInputFieldTemplate