diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-25 13:18:10 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-25 13:18:10 +0000 |
| commit | d0bf509fa14babebedbaef121ef54599003aa457 (patch) | |
| tree | 246ae35fb89dbf21fdeaf2a0296f87a70420a4c5 | |
| parent | d6ec08ba89a0e5d444b38ae80ddfe11cc7c5f4ae (diff) | |
| download | stable-diffusion-webui-gfx803-d0bf509fa14babebedbaef121ef54599003aa457.tar.gz stable-diffusion-webui-gfx803-d0bf509fa14babebedbaef121ef54599003aa457.tar.bz2 stable-diffusion-webui-gfx803-d0bf509fa14babebedbaef121ef54599003aa457.zip | |
fix for #11963
| -rw-r--r-- | extensions-builtin/Lora/networks.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/extensions-builtin/Lora/networks.py b/extensions-builtin/Lora/networks.py index 3a8cfa3b..17cbe1bb 100644 --- a/extensions-builtin/Lora/networks.py +++ b/extensions-builtin/Lora/networks.py @@ -161,7 +161,12 @@ def load_network(name, network_on_disk): sd_module = shared.sd_model.network_layer_mapping.get(key, None)
elif sd_module is None and "lora_te1_text_model" in key_network_without_network_parts:
key = key_network_without_network_parts.replace("lora_te1_text_model", "0_transformer_text_model")
- sd_module = shared.sd_model.network_layer_mapping.get(key, None) or shared.sd_model.network_layer_mapping.get(key[2:], None)
+ sd_module = shared.sd_model.network_layer_mapping.get(key, None)
+
+ # some SD1 Loras also have correct compvis keys
+ if sd_module is None:
+ key = key_network_without_network_parts.replace("lora_te1_text_model", "transformer_text_model")
+ sd_module = shared.sd_model.network_layer_mapping.get(key, None)
if sd_module is None:
keys_failed_to_match[key_network] = key
|