From 66cf65b604aed7bc26f17badb3287b2b8ab53c04 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 24 Nov 2024 13:02:45 -0500 Subject: [PATCH] correct lora assignment Signed-off-by: Vladimir Mandic --- extensions-builtin/Lora/networks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/Lora/networks.py b/extensions-builtin/Lora/networks.py index db617ee5b..fd6287c62 100644 --- a/extensions-builtin/Lora/networks.py +++ b/extensions-builtin/Lora/networks.py @@ -88,7 +88,7 @@ def assign_network_names_to_compvis_modules(sd_model): network_name = name.replace(".", "_") network_layer_mapping[network_name] = module module.network_layer_name = network_name - shared.sd_model.network_layer_mapping = network_layer_mapping + sd_model.network_layer_mapping = network_layer_mapping def load_diffusers(name, network_on_disk, lora_scale=shared.opts.extra_networks_default_multiplier) -> network.Network: @@ -141,7 +141,7 @@ def load_network(name, network_on_disk) -> network.Network: sd = sd_models.read_state_dict(network_on_disk.filename, what='network') if shared.sd_model_type == 'f1': # if kohya flux lora, convert state_dict sd = lora_convert._convert_kohya_flux_lora_to_diffusers(sd) or sd # pylint: disable=protected-access - assign_network_names_to_compvis_modules(shared.sd_model) # this should not be needed but is here as an emergency fix for an unknown error people are experiencing in 1.2.0 + assign_network_names_to_compvis_modules(shared.sd_model) keys_failed_to_match = {} matched_networks = {} bundle_embeddings = {}