diff --git a/adetailer/args.py b/adetailer/args.py index 923913f..958b149 100644 --- a/adetailer/args.py +++ b/adetailer/args.py @@ -75,7 +75,7 @@ class ADetailerArgs(BaseModel, extra=Extra.forbid): if cond: for k in pops: - p.pop(k) + p.pop(k, None) def extra_params(self, suffix: str = ""): if self.ad_model == "None": @@ -119,6 +119,9 @@ class ADetailerArgs(BaseModel, extra=Extra.forbid): ], 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) if suffix: p = {k + suffix: v for k, v in p.items()}