diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-12-14 08:54:45 +0000 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-12-14 08:54:45 +0000 |
| commit | 0fb34b57b80b368f76a368c50569371c10382e12 (patch) | |
| tree | cb32ad64211ac1670bae87946190c85abd40773e /extensions-builtin/Lora/networks.py | |
| parent | 39ebd5684b377157eca7034feec4596707cfa9c7 (diff) | |
| parent | aeaf1c510f1e7f246d892dad74122e0117a68a8c (diff) | |
| download | stable-diffusion-webui-gfx803-0fb34b57b80b368f76a368c50569371c10382e12.tar.gz stable-diffusion-webui-gfx803-0fb34b57b80b368f76a368c50569371c10382e12.tar.bz2 stable-diffusion-webui-gfx803-0fb34b57b80b368f76a368c50569371c10382e12.zip | |
Merge branch 'dev' into test-fp8
Diffstat (limited to 'extensions-builtin/Lora/networks.py')
| -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 d22ed843..985b2753 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, {})
|