Fixed issue with FreeU if no refiner is used

SeargeDP · Oct 5, 2023 · 1d3db0f · 1d3db0f
1 parent 2208f93
commit 1d3db0f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]