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

fix: fix module None issue, module not allow None

Browse Source
This commit is contained in:
Dowon
2023-10-26 21:51:38 +09:00
parent 866d9577f3
commit 96213b177f
1 changed files with 2 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
adetailer/args.py
View File

@@ -14,7 +14,6 @@ from pydantic import (
confloat,
conint,
constr,
root_validator,
validator,
)
@@ -72,20 +71,12 @@ class ADetailerArgs(BaseModel, extra=Extra.forbid):
ad_clip_skip: conint(ge=1, le=12) = 1
ad_restore_face: bool = False
ad_controlnet_model: constr(regex=cn_model_regex) = "None"
ad_controlnet_module: Optional[constr(regex=cn_module_regex)] = None
ad_controlnet_module: constr(regex=cn_module_regex) = "None"
ad_controlnet_weight: confloat(ge=0.0, le=1.0) = 1.0
ad_controlnet_guidance_start: confloat(ge=0.0, le=1.0) = 0.0
ad_controlnet_guidance_end: confloat(ge=0.0, le=1.0) = 1.0
is_api: bool = True
@root_validator(skip_on_failure=True)
def ad_controlnt_module_validator(cls, values): # noqa: N805
cn_model = values.get("ad_controlnet_model", "None")
cn_module = values.get("ad_controlnet_module", None)
if "inpaint" not in cn_model or cn_module == "None":
values["ad_controlnet_module"] = None
return values
@validator("is_api", pre=True)
def is_api_validator(cls, v: Any): # noqa: N805
"tuple is json serializable but cannot be made with json deserialize."
@@ -175,7 +166,7 @@ class ADetailerArgs(BaseModel, extra=Extra.forbid):
],
cond="None",
)
ppop("ADetailer ControlNet module")
ppop("ADetailer ControlNet module", cond="None")
ppop("ADetailer ControlNet weight", cond=1.0)
ppop("ADetailer ControlNet guidance start", cond=0.0)
ppop("ADetailer ControlNet guidance end", cond=1.0)