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

fix(ui): fix control adapters recall

Browse Source
This commit is contained in:
psychedelicious 2023-10-06 18:12:36 +11:00
parent 913fc83cbf
commit dcfbd49e1b
1 changed files with 24 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
invokeai/frontend/web/src/features/parameters/hooks/useRecallParameters.ts
View File

@ -7,6 +7,18 @@ import {
CONTROLNET_MODEL_DEFAULT_PROCESSORS,
CONTROLNET_PROCESSORS,
} from 'features/controlNet/store/constants';
import {
controlAdapterRecalled,
controlAdaptersReset,
} from 'features/controlNet/store/controlAdaptersSlice';
import {
ControlNetConfig,
IPAdapterConfig,
} from 'features/controlNet/store/types';
import {
initialControlNet,
initialIPAdapter,
} from 'features/controlNet/util/buildControlAdapter';
import {
ControlNetMetadataItem,
CoreMetadata,
@ -449,8 +461,6 @@ export const useRecallParameters = () => {
};
}
const controlNetId = uuidv4();
let processorType = initialControlNet.processorType;
for (const modelSubstring in CONTROLNET_MODEL_DEFAULT_PROCESSORS) {
if (matchingControlNetModel.model_name.includes(modelSubstring)) {
@ -463,6 +473,7 @@ export const useRecallParameters = () => {
const processorNode = CONTROLNET_PROCESSORS[processorType].default;
const controlnet: ControlNetConfig = {
type: 'controlnet',
isEnabled: true,
model: matchingControlNetModel,
weight:
@ -481,7 +492,7 @@ export const useRecallParameters = () => {
? processorNode
: initialControlNet.processorNode,
shouldAutoConfig: true,
controlNetId,
id: uuidv4(),
};
return { controlnet, error: null };
@ -498,11 +509,7 @@ export const useRecallParameters = () => {
return;
}
dispatch(
controlAdapterRecalled({
...result.controlnet,
})
);
dispatch(controlAdapterRecalled(result.controlnet));
parameterSetToast();
},
@ -558,11 +565,14 @@ export const useRecallParameters = () => {
}
const ipAdapter: IPAdapterConfig = {
adapterImage: image?.image_name ?? null,
id: uuidv4(),
type: 'ip_adapter',
isEnabled: true,
controlImage: image?.image_name ?? null,
model: matchingIPAdapterModel,
weight: weight ?? initialIPAdapterState.weight,
beginStepPct: begin_step_percent ?? initialIPAdapterState.beginStepPct,
endStepPct: end_step_percent ?? initialIPAdapterState.endStepPct,
weight: weight ?? initialIPAdapter.weight,
beginStepPct: begin_step_percent ?? initialIPAdapter.beginStepPct,
endStepPct: end_step_percent ?? initialIPAdapter.endStepPct,
};
return { ipAdapter, error: null };
@ -579,13 +589,7 @@ export const useRecallParameters = () => {
return;
}
dispatch(
ipAdapterRecalled({
...result.ipAdapter,
})
);
dispatch(isIPAdapterEnabledChanged(true));
dispatch(controlAdapterRecalled(result.ipAdapter));
parameterSetToast();
},
@ -727,9 +731,6 @@ export const useRecallParameters = () => {
});
dispatch(controlAdaptersReset());
if (controlnets?.length) {
dispatch(controlNetEnabled());
}
controlnets?.forEach((controlnet) => {
const result = prepareControlNetMetadataItem(controlnet);
if (result.controlnet) {
@ -737,13 +738,10 @@ export const useRecallParameters = () => {
}
});
if (ipAdapters?.length) {
dispatch(isIPAdapterEnabledChanged(true));
}
ipAdapters?.forEach((ipAdapter) => {
const result = prepareIPAdapterMetadataItem(ipAdapter);
if (result.ipAdapter) {
dispatch(ipAdapterRecalled(result.ipAdapter));
dispatch(controlAdapterRecalled(result.ipAdapter));
}
});