diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-12-14 10:08:03 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-12-14 10:15:36 +0300 |
| commit | 2be85f8fe03533bf3b1ad562ea58ee8227ba3b99 (patch) | |
| tree | f2dcb35b7e668d136ded04cf3f782bad7b83e9cb /extensions-builtin | |
| parent | eb52c803b849cdd1fc137db4568eca5bb8373f58 (diff) | |
Merge pull request #14237 from ReneKroon/dev
#13354 : solve lora loading issue
Diffstat (limited to 'extensions-builtin')
| -rw-r--r-- | extensions-builtin/Lora/networks.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/extensions-builtin/Lora/networks.py b/extensions-builtin/Lora/networks.py index 7f814706..629bf853 100644 --- 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, {})
|