mirror of
https://github.com/muerrilla/sd-webui-detail-daemon.git
synced 2026-01-26 11:19:46 +00:00
Merge pull request #2 from w-e-w/fix-don't-remove-all-call-back
fix don't remove all call back
This commit is contained in:
@@ -8,7 +8,7 @@ matplotlib.use('Agg')
|
||||
import matplotlib.pyplot as plt
|
||||
|
||||
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
|
||||
|
||||
|
||||
@@ -107,13 +107,13 @@ class Script(scripts.Script):
|
||||
|
||||
else:
|
||||
if hasattr(self, 'callback_added'):
|
||||
remove_current_script_callbacks()
|
||||
remove_callbacks_for_function(self.denoiser_callback)
|
||||
delattr(self, 'callback_added')
|
||||
# tqdm.write('\033[90mINFO: Detail Daemon callback removed\033[0m')
|
||||
|
||||
def postprocess(self, p, processed, *args):
|
||||
if hasattr(self, 'callback_added'):
|
||||
remove_current_script_callbacks()
|
||||
remove_callbacks_for_function(self.denoiser_callback)
|
||||
delattr(self, 'callback_added')
|
||||
# tqdm.write('\033[90mINFO: Detail Daemon callback removed\033[0m')
|
||||
|
||||
|
||||
Reference in New Issue
Block a user