Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-01-26 11:19:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: control_net_allow_script_control

Browse Source
This commit is contained in:
Bingsu
2023-05-26 11:41:25 +09:00
parent c71c8bbd53
commit 89b018dc33
2 changed files with 26 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
controlnet_ext/restore.py
View File

@@ -1,6 +1,8 @@
from __future__ import annotations
from modules import img2img, processing
from contextlib import contextmanager
from modules import img2img, processing, shared
def cn_restore_unet_hook(p, cn_latest_network):
@@ -31,3 +33,17 @@ class CNHijackRestore:
processing.process_images_inner = self.orig_process
if self.img2img:
img2img.process_batch = self.orig_img2img
@contextmanager
def cn_allow_script_control():
orig = False
if "control_net_allow_script_control" in shared.opts.data:
try:
orig = shared.opts.data["control_net_allow_script_control"]
shared.opts.data["control_net_allow_script_control"] = True
yield
finally:
shared.opts.data["control_net_allow_script_control"] = orig
else:
yield