Fix effect stackings when Reload UI

No longer keeps applying the callbacks on top of the previous ones
pull/19/head
Haoming 2023-11-14 10:26:19 +08:00
parent 89ab850c1e
commit 1e8664bf8d
1 changed files with 6 additions and 0 deletions

View File

@ -1,4 +1,5 @@
from modules.sd_samplers_kdiffusion import KDiffusionSampler
from modules import script_callbacks
from scripts.cc_scaling import apply_scaling
from scripts.cc_noise import *
@ -60,3 +61,8 @@ def cc_callback(self, d):
return original_callback(self, d)
KDiffusionSampler.callback_state = cc_callback
def restore_callback():
KDiffusionSampler.callback_state = original_callback
script_callbacks.on_script_unloaded(restore_callback)