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

feat(nodes): update LoRAMetadataItem model

Browse Source 
LoRA model now at under `model` not `lora.
This commit is contained in:
psychedelicious
2024-02-26 22:25:22 +11:00
parent a253047d8e
commit 80697a71de
5 changed files with 9 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/frontend/web/src/features/nodes/util/graph/addLoRAsToGraph.ts
View File

@ -53,7 +53,7 @@ export const addLoRAsToGraph = (
};
loraMetadata.push({
lora: { key },
model: { key },
weight,
});

13
invokeai/frontend/web/src/features/nodes/util/graph/addSDXLLoRAstoGraph.ts
View File

@ -1,8 +1,6 @@
import type { RootState } from 'app/store/store';
import type { LoRAMetadataItem } from 'features/nodes/types/metadata';
import { zLoRAMetadataItem } from 'features/nodes/types/metadata';
import { filter, size } from 'lodash-es';
import type { NonNullableGraph, SDXLLoraLoaderInvocation } from 'services/api/types';
import type { CoreMetadataInvocation, NonNullableGraph, SDXLLoraLoaderInvocation } from 'services/api/types';
import {
LORA_LOADER,
@ -36,7 +34,7 @@ export const addSDXLLoRAsToGraph = (
return;
}
const loraMetadata: LoRAMetadataItem[] = [];
const loraMetadata: CoreMetadataInvocation['loras'] = [];
// Handle Seamless Plugs
const unetLoaderId = modelLoaderNodeId;
@ -70,12 +68,7 @@ export const addSDXLLoRAsToGraph = (
weight,
};
loraMetadata.push(
zLoRAMetadataItem.parse({
lora: { key },
weight,
})
);
loraMetadata.push({ model: { key }, weight });
// add to graph
graph.nodes[currentLoraNodeId] = loraLoaderNode;