Merge branch 'dev' into extra-norm-module

1 files changed, 6 insertions, 12 deletions

diff --git a/modules/processing_scripts/refiner.py b/modules/processing_scripts/refiner.py
index 773ec5d0..3c5b37d2 100644
--- a/modules/processing_scripts/refiner.py
+++ b/modules/processing_scripts/refiner.py
@@ -38,18 +38,12 @@ class ScriptRefiner(scripts.Script):
 

         return enable_refiner, refiner_checkpoint, refiner_switch_at

 

-    def before_process(self, p, enable_refiner, refiner_checkpoint, refiner_switch_at):

+    def setup(self, p, enable_refiner, refiner_checkpoint, refiner_switch_at):

         # the actual implementation is in sd_samplers_common.py, apply_refiner

 

-        p.refiner_checkpoint_info = None

-        p.refiner_switch_at = None

-

         if not enable_refiner or refiner_checkpoint in (None, "", "None"):

-            return

-

-        refiner_checkpoint_info = sd_models.get_closet_checkpoint_match(refiner_checkpoint)

-        if refiner_checkpoint_info is None:

-            raise Exception(f'Could not find checkpoint with name {refiner_checkpoint}')

-

-        p.refiner_checkpoint_info = refiner_checkpoint_info

-        p.refiner_switch_at = refiner_switch_at

+            p.refiner_checkpoint_info = None

+            p.refiner_switch_at = None

+        else:

+            p.refiner_checkpoint = refiner_checkpoint

+            p.refiner_switch_at = refiner_switch_at