diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-12-14 10:08:03 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-14 10:08:03 +0300 |
| commit | 778a30a95e216f9f7ce0126dbbdb1334afc3d796 (patch) | |
| tree | 8d5fcc86015758f95ea5ccd0a8ac115a76f2b32d /extensions-builtin | |
| parent | 96c393a7a77d8610abdba39197964be64eedef02 (diff) | |
| parent | 16bdcce92d5b482d50cdc32a8f308040d320b6c9 (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, {})
|