also use setup callback for the refiner instead of before_process

AUTOMATIC1111 · Aug 13, 2023 · 007ecfb · 007ecfb
1 parent 9cd0475
commit 007ecfb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/processing_scripts/refiner.py b/modules/processing_scripts/refiner.py
@@ -38,7 +38,7 @@ def lookup_checkpoint(title):
 
         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
 
         if not enable_refiner or refiner_checkpoint in (None, "", "None"):