also use setup callback for the refiner instead of before_process

This commit is contained in:
AUTOMATIC1111 2023-08-13 21:01:13 +03:00
parent 9cd0475c08
commit 007ecfbb29

View File

@ -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"):