muerrilla/sd-webui-detail-daemon
1
0
Fork 0
mirror of https://github.com/muerrilla/sd-webui-detail-daemon.git synced 2026-01-26 19:29:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix don't remove all call back

Browse Source
This commit is contained in:
w-e-w
2024-05-15 19:43:23 +09:00
parent 447b915e93
commit 9d0be13cfe
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/detail_daemon.py
View File

@@ -8,7 +8,7 @@ matplotlib.use('Agg')
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import modules.scripts as scripts import modules.scripts as scripts
from modules.script_callbacks import on_cfg_denoiser, remove_current_script_callbacks, on_infotext_pasted from modules.script_callbacks import on_cfg_denoiser, remove_callbacks_for_function, on_infotext_pasted
from modules.ui_components import InputAccordion from modules.ui_components import InputAccordion
@@ -107,13 +107,13 @@ class Script(scripts.Script):
else: else:
if hasattr(self, 'callback_added'): if hasattr(self, 'callback_added'):
remove_current_script_callbacks() remove_callbacks_for_function(self.denoiser_callback)
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 postprocess(self, p, processed, *args): def postprocess(self, p, processed, *args):
if hasattr(self, 'callback_added'): if hasattr(self, 'callback_added'):
remove_current_script_callbacks() remove_callbacks_for_function(self.denoiser_callback)
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')