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]