Commit 007ecfbb authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

also use setup callback for the refiner instead of before_process

parent 9cd0475c
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -38,7 +38,7 @@ 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

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