fix disable during hires fix

This commit is contained in:
w-e-w
2024-07-01 22:27:12 +09:00
parent 9621682aee
commit da12d2edc5

View File

@@ -123,7 +123,6 @@ class Script(scripts.Script):
actual_steps = (p.steps * 2 - 1) if p.sampler_name in ['DPM++ SDE', 'DPM++ 2S a', 'Heun', 'DPM2', 'DPM2 a', 'Restart'] else p.steps
self.schedule = self.make_schedule(actual_steps, start, end, bias, amount, exponent, start_offset, end_offset, fade, smooth)
self.mode = mode
self.is_hires = False
self.cfg_scale = p.cfg_scale
on_cfg_denoiser(self.denoiser_callback)
self.callback_added = True
@@ -135,6 +134,9 @@ class Script(scripts.Script):
delattr(self, 'callback_added')
# tqdm.write('\033[90mINFO: Detail Daemon callback removed\033[0m')
def before_process_batch(self, p, *args, **kwargs):
self.is_hires = False
def postprocess(self, p, processed, *args):
if hasattr(self, 'callback_added'):
remove_callbacks_for_function(self.denoiser_callback)