fixes

DrakeRichards · Dec 3, 2023 · b59f796 · b59f796
1 parent 323e2c1
commit b59f796
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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]