aboutsummaryrefslogtreecommitdiff
path: root/extensions-builtin
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-07-25 16:17:41 +0300
committerGitHub <noreply@github.com>2023-07-25 16:17:41 +0300
commitd6ec08ba89a0e5d444b38ae80ddfe11cc7c5f4ae (patch)
tree01278fb4a58e04f895d6f961525249566ebebf02 /extensions-builtin
parent65bf3ba260ab3435b574c615ebff21defb9bf89f (diff)
parenta68f4690307b0f94404b85513429e18ee3936589 (diff)
Merge pull request #11963 from catboxanon/fix/lora-te
Fix parsing text encoder blocks in some LoRAs
Diffstat (limited to 'extensions-builtin')
-rw-r--r--extensions-builtin/Lora/networks.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/extensions-builtin/Lora/networks.py b/extensions-builtin/Lora/networks.py
index af8188e3..3a8cfa3b 100644
--- a/extensions-builtin/Lora/networks.py
+++ b/extensions-builtin/Lora/networks.py
@@ -161,7 +161,7 @@ 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)
+ sd_module = shared.sd_model.network_layer_mapping.get(key, None) or shared.sd_model.network_layer_mapping.get(key[2:], None)
if sd_module is None:
keys_failed_to_match[key_network] = key