mirror of
https://github.com/muerrilla/sd-webui-detail-daemon.git
synced 2026-01-26 11:19:46 +00:00
fix disable during hires fix
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user