Skip to content

Commit b57a816

Browse files
committed
Merge pull request AUTOMATIC1111#11963 from catboxanon/fix/lora-te
Fix parsing text encoder blocks in some LoRAs
1 parent 11f996a commit b57a816

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

extensions-builtin/Lora/networks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@ def load_network(name, network_on_disk):
161161
sd_module = shared.sd_model.network_layer_mapping.get(key, None)
162162
elif sd_module is None and "lora_te1_text_model" in key_network_without_network_parts:
163163
key = key_network_without_network_parts.replace("lora_te1_text_model", "0_transformer_text_model")
164-
sd_module = shared.sd_model.network_layer_mapping.get(key, None)
164+
sd_module = shared.sd_model.network_layer_mapping.get(key, None) or shared.sd_model.network_layer_mapping.get(key[2:], None)
165165

166166
if sd_module is None:
167167
keys_failed_to_match[key_network] = key

0 commit comments

Comments
 (0)