Skip to content

Commit f057d5c

Browse files
fix: Lint Errors
1 parent be00a83 commit f057d5c

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ import { selectIsBusy } from 'features/system/store/systemSelectors';
1212
import { forEach } from 'lodash-es';
1313
import { memo, useCallback, useMemo } from 'react';
1414
import { useGetControlNetModelsQuery } from 'services/api/endpoints/models';
15+
import { BaseModelType } from 'services/api/types';
1516

1617
type ParamControlNetModelProps = {
1718
controlNetId: string;
@@ -58,7 +59,7 @@ const ParamControlNetModel = (props: ParamControlNetModelProps) => {
5859
data.push({
5960
value: id,
6061
label: model.model_name,
61-
group: MODEL_TYPE_MAP[model.base_model],
62+
group: MODEL_TYPE_MAP[model.base_model as BaseModelType],
6263
disabled,
6364
tooltip: disabled
6465
? `Incompatible base model: ${model.base_model}`

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

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ import { modelIdToControlNetModelParam } from 'features/parameters/util/modelIdT
1111
import { forEach } from 'lodash-es';
1212
import { memo, useCallback, useMemo } from 'react';
1313
import { useGetControlNetModelsQuery } from 'services/api/endpoints/models';
14+
import { BaseModelType } from 'services/api/types';
1415
import { FieldComponentProps } from './types';
1516

1617
const ControlNetModelInputFieldComponent = (
@@ -53,7 +54,7 @@ const ControlNetModelInputFieldComponent = (
5354
data.push({
5455
value: id,
5556
label: model.model_name,
56-
group: MODEL_TYPE_MAP[model.base_model],
57+
group: MODEL_TYPE_MAP[model.base_model as BaseModelType],
5758
});
5859
});
5960

@@ -87,7 +88,8 @@ const ControlNetModelInputFieldComponent = (
8788
<IAIMantineSelect
8889
tooltip={selectedModel?.description}
8990
label={
90-
selectedModel?.base_model && MODEL_TYPE_MAP[selectedModel?.base_model]
91+
selectedModel?.base_model &&
92+
MODEL_TYPE_MAP[selectedModel?.base_model as BaseModelType]
9193
}
9294
value={selectedModel?.id ?? null}
9395
placeholder="Pick one"

0 commit comments

Comments
 (0)