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): filter disabled LoRAs on sdxl

Browse Source
This commit is contained in:
psychedelicious 2024-02-09 11:12:07 +11:00 committed by Kent Keirsey
parent a68d8fe203
commit c65d497cbc
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,7 +1,7 @@
import type { RootState } from 'app/store/store'; import type { RootState } from 'app/store/store';
import type { LoRAMetadataItem } from 'features/nodes/types/metadata'; import type { LoRAMetadataItem } from 'features/nodes/types/metadata';
import { zLoRAMetadataItem } from 'features/nodes/types/metadata'; import { zLoRAMetadataItem } from 'features/nodes/types/metadata';
import { forEach, size } from 'lodash-es'; import { filter, size } from 'lodash-es';
import type { NonNullableGraph, SDXLLoraLoaderInvocation } from 'services/api/types'; import type { NonNullableGraph, SDXLLoraLoaderInvocation } from 'services/api/types';
import { import {
@ -31,8 +31,8 @@ export const addSDXLLoRAsToGraph = (
* So we need to inject a LoRA chain into the graph. * So we need to inject a LoRA chain into the graph.
*/ */
const { loras } = state.lora; const enabledLoRAs = filter(state.lora.loras, (l) => l.isEnabled ?? false);
const loraCount = size(loras); const loraCount = size(enabledLoRAs);
if (loraCount === 0) { if (loraCount === 0) {
return; return;
@ -59,7 +59,7 @@ export const addSDXLLoRAsToGraph = (
let lastLoraNodeId = ''; let lastLoraNodeId = '';
let currentLoraIndex = 0; let currentLoraIndex = 0;
forEach(loras, (lora) => { enabledLoRAs.forEach((lora) => {
const { model_name, base_model, weight } = lora; const { model_name, base_model, weight } = lora;
const currentLoraNodeId = `${LORA_LOADER}_${model_name.replace('.', '_')}`; const currentLoraNodeId = `${LORA_LOADER}_${model_name.replace('.', '_')}`;