AUTOMATIC1111#13354 Fix lora loading

ReneKroon · Dec 7, 2023 · 7a77055 · 7a77055
1 parent 48fae7c
commit 7a77055
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/extensions-builtin/Lora/networks.py b/extensions-builtin/Lora/networks.py
@@ -159,7 +159,11 @@ def load_network(name, network_on_disk):
     bundle_embeddings = {}
 
     for key_network, weight in sd.items():
-        key_network_without_network_parts, network_part = key_network.split(".", 1)
+        if "." in key_network:
+            key_network_without_network_parts, network_part = key_network.split(".", 1)
+        else:
+            key_network_without_network_parts = key_network
+            network_part = None
         if key_network_without_network_parts == "bundle_emb":
             emb_name, vec_name = network_part.split(".", 1)
             emb_dict = bundle_embeddings.get(emb_name, {})