File tree Expand file tree Collapse file tree 2 files changed +6
-3
lines changed
invokeai/frontend/web/src/features
controlNet/components/parameters Expand file tree Collapse file tree 2 files changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -12,6 +12,7 @@ import { selectIsBusy } from 'features/system/store/systemSelectors';
1212import { forEach } from 'lodash-es' ;
1313import { memo , useCallback , useMemo } from 'react' ;
1414import { useGetControlNetModelsQuery } from 'services/api/endpoints/models' ;
15+ import { BaseModelType } from 'services/api/types' ;
1516
1617type 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 } `
Original file line number Diff line number Diff line change @@ -11,6 +11,7 @@ import { modelIdToControlNetModelParam } from 'features/parameters/util/modelIdT
1111import { forEach } from 'lodash-es' ;
1212import { memo , useCallback , useMemo } from 'react' ;
1313import { useGetControlNetModelsQuery } from 'services/api/endpoints/models' ;
14+ import { BaseModelType } from 'services/api/types' ;
1415import { FieldComponentProps } from './types' ;
1516
1617const 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"
You can’t perform that action at this time.
0 commit comments