From cd71fe51ff0a36794f9c85a6584eb67bdce19ed7 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Tue, 25 Mar 2025 14:56:53 -0400 Subject: [PATCH] fix lora change with balanced offload Signed-off-by: Vladimir Mandic --- modules/lora/extra_networks_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/lora/extra_networks_lora.py b/modules/lora/extra_networks_lora.py index 8aeb3f473..526926c9a 100644 --- a/modules/lora/extra_networks_lora.py +++ b/modules/lora/extra_networks_lora.py @@ -166,12 +166,12 @@ class ExtraNetworkLora(extra_networks.ExtraNetwork): if has_changed: networks.network_deactivate(include, exclude) networks.network_activate(include, exclude) + shared.sd_model = sd_models.apply_balanced_offload(shared.sd_model) # TODO lora: required for flux to reapply offload after lora has been applied, but fails with oom debug_log(f'Network load: type=LoRA previous={[n.name for n in networks.previously_loaded_networks]} current={[n.name for n in networks.loaded_networks]} changed') if len(networks.loaded_networks) > 0 and (len(networks.applied_layers) > 0 or force_diffusers) and step == 0: infotext(p) prompt(p) - shared.sd_model = sd_models.apply_balanced_offload(shared.sd_model) # TODO lora: required for flux to reapply offload after lora has been applied, but fails with oom if (has_changed or force_diffusers) and len(include) == 0: # print only once shared.log.info(f'Network load: type=LoRA apply={[n.name for n in networks.loaded_networks]} mode={"fuse" if shared.opts.lora_fuse_diffusers else "backup"} te={te_multipliers} unet={unet_multipliers} time={networks.timer.summary}')