diff options
| author | Rene Kroon <kroon.r.w@gmail.com> | 2023-12-08 21:19:29 +0100 |
|---|---|---|
| committer | Rene Kroon <kroon.r.w@gmail.com> | 2023-12-08 21:20:55 +0100 |
| commit | 16bdcce92d5b482d50cdc32a8f308040d320b6c9 (patch) | |
| tree | e13429b1dacaeb868296e08d701b950fd1787448 /extensions-builtin | |
| parent | f92d61497a426a19818625c3ccdaae9beeb82b31 (diff) | |
#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, {})
|