diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-12-16 07:05:10 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-12-16 07:05:10 +0000 |
| commit | e9c6325fc635302e2b4b8295345833cb8b15f7fb (patch) | |
| tree | b501070bcf1ae667b1555e682598973818c945e0 /extensions-builtin/Lora/networks.py | |
| parent | 29f04149b60bcf6e8e2b41a161d6cc7e8981710f (diff) | |
| parent | 7504f14503d6ce5a014f1c558ea4f4d57675e1e9 (diff) | |
| download | stable-diffusion-webui-gfx803-e9c6325fc635302e2b4b8295345833cb8b15f7fb.tar.gz stable-diffusion-webui-gfx803-e9c6325fc635302e2b4b8295345833cb8b15f7fb.tar.bz2 stable-diffusion-webui-gfx803-e9c6325fc635302e2b4b8295345833cb8b15f7fb.zip | |
Merge branch 'dev' into torch210
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 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, {})
|