diff --git a/scripts/detail_daemon.py b/scripts/detail_daemon.py index b269317..ce6ee4d 100644 --- a/scripts/detail_daemon.py +++ b/scripts/detail_daemon.py @@ -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)