diff options
| author | catboxanon <122327233+catboxanon@users.noreply.github.com> | 2023-07-24 17:54:59 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-24 17:54:59 -0400 |
| commit | a68f4690307b0f94404b85513429e18ee3936589 (patch) | |
| tree | ce22d5e2f43c6cff9596cbc0df13c33373dd97be /extensions-builtin/Lora | |
| parent | f7c0a963f15b165c8518faad87ab10323baa173f (diff) | |
Fix to parse TE in some LoRAs
Diffstat (limited to 'extensions-builtin/Lora')
| -rw-r--r-- | extensions-builtin/Lora/networks.py | 2 |
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
|