From bd0878754c70542caf9a417e534fbd7a65817d93 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Fri, 9 Feb 2024 21:40:19 -0800 Subject: [PATCH] apply loras to refiner like upstream #161 --- modules/sd_samplers_common.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py index a442e150..77ae3812 100644 --- a/modules/sd_samplers_common.py +++ b/modules/sd_samplers_common.py @@ -6,6 +6,7 @@ from PIL import Image from modules import devices, images, sd_vae_approx, sd_samplers, sd_vae_taesd, shared, sd_models from modules.shared import opts, state from modules_forge.forge_sampler import sampling_prepare, sampling_cleanup +from modules import extra_networks import k_diffusion.sampling @@ -182,7 +183,9 @@ def apply_refiner(cfg_denoiser, x): with sd_models.SkipWritingToConfig(): sd_models.reload_model_weights(info=refiner_checkpoint_info) - devices.torch_gc() + if not cfg_denoiser.p.disable_extra_networks: + extra_networks.activate(cfg_denoiser.p, cfg_denoiser.p.extra_network_data) + cfg_denoiser.p.setup_conds() cfg_denoiser.update_inner_model()