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

fix: fix controlnet module regex

Browse Source
This commit is contained in:
Dowon
2023-10-24 20:15:43 +09:00
parent 10fd2906a3
commit 3c8e8223ae
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
adetailer/args.py
View File

@@ -19,6 +19,7 @@ from pydantic import (
)
cn_model_regex = r".*(inpaint|tile|scribble|lineart|openpose).*|^None$"
cn_module_regex = r".*(inpaint|tile|pidi|lineart|openpose).*|^None$"
class Arg(NamedTuple):
@@ -71,7 +72,7 @@ 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_model_regex)] = None
ad_controlnet_module: Optional[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