From b8b46aec09c84b3f1450468563ed72ffe526c69f Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Fri, 28 Jul 2023 04:34:41 +1200
Subject: [PATCH] Revert "fix: Lint Errors"

This reverts commit f057d5c85b5a0e3434ae9ad27fb4ca68f50500d5.
---
 .../components/parameters/ParamControlNetModel.tsx          | 3 +--
 .../fields/ControlNetModelInputFieldComponent.tsx           | 6 ++----
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/invokeai/frontend/web/src/features/controlNet/components/parameters/ParamControlNetModel.tsx b/invokeai/frontend/web/src/features/controlNet/components/parameters/ParamControlNetModel.tsx
index 9de9afbffb..8392bdd2e3 100644
--- a/invokeai/frontend/web/src/features/controlNet/components/parameters/ParamControlNetModel.tsx
+++ b/invokeai/frontend/web/src/features/controlNet/components/parameters/ParamControlNetModel.tsx
@@ -12,7 +12,6 @@ import { selectIsBusy } from 'features/system/store/systemSelectors';
 import { forEach } from 'lodash-es';
 import { memo, useCallback, useMemo } from 'react';
 import { useGetControlNetModelsQuery } from 'services/api/endpoints/models';
-import { BaseModelType } from 'services/api/types';
 
 type ParamControlNetModelProps = {
   controlNetId: string;
@@ -59,7 +58,7 @@ const ParamControlNetModel = (props: ParamControlNetModelProps) => {
       data.push({
         value: id,
         label: model.model_name,
-        group: MODEL_TYPE_MAP[model.base_model as BaseModelType],
+        group: MODEL_TYPE_MAP[model.base_model],
         disabled,
         tooltip: disabled
           ? `Incompatible base model: ${model.base_model}`
diff --git a/invokeai/frontend/web/src/features/nodes/components/fields/ControlNetModelInputFieldComponent.tsx b/invokeai/frontend/web/src/features/nodes/components/fields/ControlNetModelInputFieldComponent.tsx
index 7d45f1c1d0..7fe2373f66 100644
--- a/invokeai/frontend/web/src/features/nodes/components/fields/ControlNetModelInputFieldComponent.tsx
+++ b/invokeai/frontend/web/src/features/nodes/components/fields/ControlNetModelInputFieldComponent.tsx
@@ -11,7 +11,6 @@ import { modelIdToControlNetModelParam } from 'features/parameters/util/modelIdT
 import { forEach } from 'lodash-es';
 import { memo, useCallback, useMemo } from 'react';
 import { useGetControlNetModelsQuery } from 'services/api/endpoints/models';
-import { BaseModelType } from 'services/api/types';
 import { FieldComponentProps } from './types';
 
 const ControlNetModelInputFieldComponent = (
@@ -54,7 +53,7 @@ const ControlNetModelInputFieldComponent = (
       data.push({
         value: id,
         label: model.model_name,
-        group: MODEL_TYPE_MAP[model.base_model as BaseModelType],
+        group: MODEL_TYPE_MAP[model.base_model],
       });
     });
 
@@ -88,8 +87,7 @@ const ControlNetModelInputFieldComponent = (
     <IAIMantineSelect
       tooltip={selectedModel?.description}
       label={
-        selectedModel?.base_model &&
-        MODEL_TYPE_MAP[selectedModel?.base_model as BaseModelType]
+        selectedModel?.base_model && MODEL_TYPE_MAP[selectedModel?.base_model]
       }
       value={selectedModel?.id ?? null}
       placeholder="Pick one"