From 1d3db0f69715635ad9b86e15dcc06ef7768928b0 Mon Sep 17 00:00:00 2001 From: SeargeDP <searge@ocean-labs.de> Date: Thu, 5 Oct 2023 02:11:40 +0200 Subject: [PATCH] Fixed issue with FreeU if no refiner is used --- modules/stage_load_checkpoints.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/stage_load_checkpoints.py b/modules/stage_load_checkpoints.py index 8c99b42..7fce086 100644 --- a/modules/stage_load_checkpoints.py +++ b/modules/stage_load_checkpoints.py @@ -136,7 +136,7 @@ def process(self, data, stage_input): refiner_checkpoint = access.get_from_cache(Names.C_REFINER_CHECKPOINT) refiner_model = refiner_checkpoint[0] - if freeu_mode is not None and freeu_mode != UI.NONE: + if freeu_mode is not None and freeu_mode != UI.NONE and refiner_model is not None: refiner_model = NodeWrapper.freeu.patch(refiner_model, b1, b2, s1, s2)[0] refiner_clip = refiner_checkpoint[1] refiner_vae = refiner_checkpoint[2]