From b59f79657d88267400eb08c1a2c63e73834df43b Mon Sep 17 00:00:00 2001 From: AI-Casanova <54461896+AI-Casanova@users.noreply.github.com> Date: Sun, 3 Dec 2023 17:20:44 -0600 Subject: [PATCH] fixes --- extensions-builtin/Lora/extra_networks_lora.py | 2 +- extensions-builtin/Lora/network.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/extensions-builtin/Lora/extra_networks_lora.py b/extensions-builtin/Lora/extra_networks_lora.py index c9995013c15..a5d9e172602 100644 --- a/extensions-builtin/Lora/extra_networks_lora.py +++ b/extensions-builtin/Lora/extra_networks_lora.py @@ -36,7 +36,7 @@ def activate(self, p, params_list): te_multiplier = float(params.positional[1]) if len(params.positional) > 1 else 1.0 te_multiplier = float(params.named.get("te", te_multiplier)) unet_multiplier = [float(params.positional[2]) if len(params.positional) > 2 else te_multiplier] * 3 - unet_multiplier = [float(params.named.get("unet", unet_multiplier))] * 3 + unet_multiplier = [float(params.named.get("unet", unet_multiplier[0]))] * 3 unet_multiplier[0] = float(params.named.get("in", unet_multiplier[0])) unet_multiplier[1] = float(params.named.get("mid", unet_multiplier[1])) unet_multiplier[2] = float(params.named.get("out", unet_multiplier[2])) diff --git a/extensions-builtin/Lora/network.py b/extensions-builtin/Lora/network.py index 8445fb36125..e5828daf3f2 100644 --- a/extensions-builtin/Lora/network.py +++ b/extensions-builtin/Lora/network.py @@ -112,11 +112,11 @@ def __init__(self, net: Network, weights: NetworkWeights): def multiplier(self): if 'transformer' in self.sd_key[:20]: return self.network.te_multiplier - if "input_blocks" in self.sd_key: + if "down_blocks" in self.sd_key: return self.network.unet_multiplier[0] - if "middle_block" in self.sd_key: + if "mid_block" in self.sd_key: return self.network.unet_multiplier[1] - if "output_blocks" in self.sd_key: + if "up_blocks" in self.sd_key: return self.network.unet_multiplier[2] else: return self.network.unet_multiplier[0]