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

Merge branch 'main' into maryhipp/informational-popover

Browse Source
This commit is contained in:
Kent Keirsey 2023-09-20 13:32:06 -04:00 committed by GitHub
commit f4c895257a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
invokeai/frontend/web/src/features/parameters/components/Parameters/ControlNet/ParamControlNetCollapse.tsx
View File

@ -15,12 +15,9 @@ import {
import { getValidControlNets } from 'features/controlNet/util/getValidControlNets';
import { useFeatureStatus } from 'features/system/hooks/useFeatureStatus';
import { map } from 'lodash-es';
import { Fragment, memo, useCallback } from 'react';
import { Fragment, memo, useCallback, useMemo } from 'react';
import { FaPlus } from 'react-icons/fa';
import {
controlNetModelsAdapter,
useGetControlNetModelsQuery,
} from 'services/api/endpoints/models';
import { useGetControlNetModelsQuery } from 'services/api/endpoints/models';
import { v4 as uuidv4 } from 'uuid';
const selector = createSelector(
@ -53,16 +50,22 @@ const ParamControlNetCollapse = () => {
const { controlNetsArray, activeLabel } = useAppSelector(selector);
const isControlNetDisabled = useFeatureStatus('controlNet').isFeatureDisabled;
const dispatch = useAppDispatch();
const { firstModel } = useGetControlNetModelsQuery(undefined, {
selectFromResult: (result) => {
const firstModel = result.data
? controlNetModelsAdapter.getSelectors().selectAll(result.data)[0]
: undefined;
return {
firstModel,
};
},
});
const { data: controlnetModels } = useGetControlNetModelsQuery();
const firstModel = useMemo(() => {
if (!controlnetModels || !Object.keys(controlnetModels.entities).length) {
return undefined;
}
const firstModelId = Object.keys(controlnetModels.entities)[0];
if (!firstModelId) {
return undefined;
}
const firstModel = controlnetModels.entities[firstModelId];
return firstModel ? firstModel : undefined;
}, [controlnetModels]);
const handleClickedAddControlNet = useCallback(() => {
if (!firstModel) {