1580158
/
adetailer
mirror of https://github.com/Bing-su/adetailer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(scripts): copy scripts args 

maybe fix #701
pull/728/head
Dowon 2024-09-09 22:08:25 +09:00
parent 17694f9ac2
commit ceacc711e4
No known key found for this signature in database
GPG Key ID: 31D3079F422E1444
1 changed files with 20 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
scripts/!adetailer.py
View File

@ -474,18 +474,26 @@ class AfterDetailerScript(scripts.Script):
script_runner.alwayson_scripts = filtered_alwayson script_runner.alwayson_scripts = filtered_alwayson
return script_runner, script_args return script_runner, script_args
def disable_controlnet_units( def disable_controlnet_units(self, script_args: Sequence[Any]) -> list[Any]:
self, script_args: list[Any] | tuple[Any, ...] new_args = []
) -> None: for arg in script_args:
for obj in script_args: if "controlnet" in arg.__class__.__name__.lower():
if "controlnet" in obj.__class__.__name__.lower(): new = copy(arg)
if hasattr(obj, "enabled"): if hasattr(new, "enabled"):
obj.enabled = False new.enabled = False
if hasattr(obj, "input_mode"): if hasattr(new, "input_mode"):
obj.input_mode = getattr(obj.input_mode, "SIMPLE", "simple") new.input_mode = getattr(new.input_mode, "SIMPLE", "simple")
new_args.append(new)
elif isinstance(obj, dict) and "module" in obj: elif isinstance(arg, dict) and "module" in arg:
obj["enabled"] = False new = copy(arg)
new["enabled"] = False
new_args.append(new)
else:
new_args.append(arg)
return new_args
def get_i2i_p(self, p, args: ADetailerArgs, image): def get_i2i_p(self, p, args: ADetailerArgs, image):
seed, subseed = self.get_seed(p) seed, subseed = self.get_seed(p)
@ -545,7 +553,7 @@ class AfterDetailerScript(scripts.Script):
i2i._ad_inner = True i2i._ad_inner = True
if args.ad_controlnet_model != "Passthrough" and controlnet_type != "forge": if args.ad_controlnet_model != "Passthrough" and controlnet_type != "forge":
self.disable_controlnet_units(i2i.script_args) i2i.script_args = self.disable_controlnet_units(i2i.script_args)
if args.ad_controlnet_model not in ["None", "Passthrough"]: if args.ad_controlnet_model not in ["None", "Passthrough"]:
self.update_controlnet_args(i2i, args) self.update_controlnet_args(i2i, args)