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: Lint Errors

Browse Source
This commit is contained in:
blessedcoolant 2023-07-27 19:40:32 +12:00
parent be00a837cc
commit f057d5c85b
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/frontend/web/src/features/controlNet/components/parameters/ParamControlNetModel.tsx
View File

@ -12,6 +12,7 @@ import { selectIsBusy } from 'features/system/store/systemSelectors';
import { forEach } from 'lodash-es'; import { forEach } from 'lodash-es';
import { memo, useCallback, useMemo } from 'react'; import { memo, useCallback, useMemo } from 'react';
import { useGetControlNetModelsQuery } from 'services/api/endpoints/models'; import { useGetControlNetModelsQuery } from 'services/api/endpoints/models';
import { BaseModelType } from 'services/api/types';
type ParamControlNetModelProps = { type ParamControlNetModelProps = {
controlNetId: string; controlNetId: string;
@ -58,7 +59,7 @@ const ParamControlNetModel = (props: ParamControlNetModelProps) => {
data.push({ data.push({
value: id, value: id,
label: model.model_name, label: model.model_name,
group: MODEL_TYPE_MAP[model.base_model], group: MODEL_TYPE_MAP[model.base_model as BaseModelType],
disabled, disabled,
tooltip: disabled tooltip: disabled
? `Incompatible base model: ${model.base_model}` ? `Incompatible base model: ${model.base_model}`

6
invokeai/frontend/web/src/features/nodes/components/fields/ControlNetModelInputFieldComponent.tsx
View File

@ -11,6 +11,7 @@ import { modelIdToControlNetModelParam } from 'features/parameters/util/modelIdT
import { forEach } from 'lodash-es'; import { forEach } from 'lodash-es';
import { memo, useCallback, useMemo } from 'react'; import { memo, useCallback, useMemo } from 'react';
import { useGetControlNetModelsQuery } from 'services/api/endpoints/models'; import { useGetControlNetModelsQuery } from 'services/api/endpoints/models';
import { BaseModelType } from 'services/api/types';
import { FieldComponentProps } from './types'; import { FieldComponentProps } from './types';
const ControlNetModelInputFieldComponent = ( const ControlNetModelInputFieldComponent = (
@ -53,7 +54,7 @@ const ControlNetModelInputFieldComponent = (
data.push({ data.push({
value: id, value: id,
label: model.model_name, label: model.model_name,
group: MODEL_TYPE_MAP[model.base_model], group: MODEL_TYPE_MAP[model.base_model as BaseModelType],
}); });
}); });
@ -87,7 +88,8 @@ const ControlNetModelInputFieldComponent = (
<IAIMantineSelect <IAIMantineSelect
tooltip={selectedModel?.description} tooltip={selectedModel?.description}
label={ label={
selectedModel?.base_model && MODEL_TYPE_MAP[selectedModel?.base_model] selectedModel?.base_model &&
MODEL_TYPE_MAP[selectedModel?.base_model as BaseModelType]
} }
value={selectedModel?.id ?? null} value={selectedModel?.id ?? null}
placeholder="Pick one" placeholder="Pick one"