muerrilla/sd-webui-detail-daemon
1
0
Fork 0
mirror of https://github.com/muerrilla/sd-webui-detail-daemon.git synced 2026-05-01 03:31:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix disable during hires fix

Browse Source
This commit is contained in:
w-e-w
2024-07-01 22:27:12 +09:00
parent 9621682aee
commit da12d2edc5
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/detail_daemon.py
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 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.schedule = self.make_schedule(actual_steps, start, end, bias, amount, exponent, start_offset, end_offset, fade, smooth)
self.mode = mode self.mode = mode
self.is_hires = False
self.cfg_scale = p.cfg_scale self.cfg_scale = p.cfg_scale
on_cfg_denoiser(self.denoiser_callback) on_cfg_denoiser(self.denoiser_callback)
self.callback_added = True self.callback_added = True
@@ -135,6 +134,9 @@ class Script(scripts.Script):
delattr(self, 'callback_added') delattr(self, 'callback_added')
# tqdm.write('\033[90mINFO: Detail Daemon callback removed\033[0m') # 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): def postprocess(self, p, processed, *args):
if hasattr(self, 'callback_added'): if hasattr(self, 'callback_added'):
remove_callbacks_for_function(self.denoiser_callback) remove_callbacks_for_function(self.denoiser_callback)