Fix store not resetting.

Ping #25
This commit is contained in:
shiimizu
2024-09-06 09:56:11 -07:00
parent 217aa13d72
commit 5b2d0d2c40
+8
View File
@@ -28,6 +28,9 @@ def KSAMPLER_sample(*args, **kwargs):
store.sigmas = sigmas
store.model_options = model_options
store.extra_args = extra_args
else:
for attr in ['sigmas', 'model_options', 'extra_args']:
_delattr(store, attr)
return orig_fn(*args, **kwargs)
def KSampler_sample(*args, **kwargs):
@@ -62,6 +65,11 @@ def get_area_and_mult(*args, **kwargs):
control.get_control = control.get_control_orig
return store.get_area_and_mult(*args, **kwargs)
def _delattr(obj, attr):
try:
if hasattr(obj, attr): delattr(obj, attr)
except Exception: ...
def register_hooks():
patches = [
(comfy.samplers.KSampler, 'sample', KSampler_sample),