Merge pull request #14237 from ReneKroon/dev

#13354 : solve lora loading issue
2023-12-14 10:08:03 +03:00 · 2023-12-14 10:08:03 +03:00 · 2be85f8fe0
parent eb52c803b8
commit 2be85f8fe0
1 changed files with 2 additions and 1 deletions
--- a/extensions-builtin/Lora/networks.py
+++ b/extensions-builtin/Lora/networks.py
@ -159,7 +159,8 @@ 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)
+        key_network_without_network_parts, _, network_part = key_network.partition(".")
+
        if key_network_without_network_parts == "bundle_emb":
            emb_name, vec_name = network_part.split(".", 1)
            emb_dict = bundle_embeddings.get(emb_name, {})