From 610891f7295bb70476dc0b644d5a640ca95002dc Mon Sep 17 00:00:00 2001 From: tobias17 Date: Mon, 10 Apr 2023 20:03:32 -0400 Subject: [PATCH] fixed lora unpack bug --- extensions-builtin/Lora/lora.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/extensions-builtin/Lora/lora.py b/extensions-builtin/Lora/lora.py index d3eb0d3bc..22076a330 100644 --- a/extensions-builtin/Lora/lora.py +++ b/extensions-builtin/Lora/lora.py @@ -136,7 +136,10 @@ def load_lora(name, filename): is_sd2 = 'model_transformer_resblocks' in shared.sd_model.lora_layer_mapping for key_diffusers, weight in sd.items(): - key_diffusers_without_lora_parts, lora_key = key_diffusers.split(".", 1) + lora_key_parts = key_diffusers.split(".", 1) + key_diffusers_without_lora_parts = lora_key_parts[0] + lora_key = lora_key_parts[1] if len(lora_key_parts) > 1 else "" + key = convert_diffusers_name_to_compvis(key_diffusers_without_lora_parts, is_sd2) sd_module = shared.sd_model.lora_layer_mapping.get(key, None)