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 8392bdd2e3..9de9afbffb 100644
--- a/invokeai/frontend/web/src/features/controlNet/components/parameters/ParamControlNetModel.tsx
+++ b/invokeai/frontend/web/src/features/controlNet/components/parameters/ParamControlNetModel.tsx
@@ -12,6 +12,7 @@ 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;
@@ -58,7 +59,7 @@ const ParamControlNetModel = (props: ParamControlNetModelProps) => {
       data.push({
         value: id,
         label: model.model_name,
-        group: MODEL_TYPE_MAP[model.base_model],
+        group: MODEL_TYPE_MAP[model.base_model as BaseModelType],
         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 7fe2373f66..7d45f1c1d0 100644
--- a/invokeai/frontend/web/src/features/nodes/components/fields/ControlNetModelInputFieldComponent.tsx
+++ b/invokeai/frontend/web/src/features/nodes/components/fields/ControlNetModelInputFieldComponent.tsx
@@ -11,6 +11,7 @@ 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 = (
@@ -53,7 +54,7 @@ const ControlNetModelInputFieldComponent = (
       data.push({
         value: id,
         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
       tooltip={selectedModel?.description}
       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}
       placeholder="Pick one"