diff --git a/scripts/posex.py b/scripts/posex.py index 2cb5ff5..89a389b 100644 --- a/scripts/posex.py +++ b/scripts/posex.py @@ -77,7 +77,7 @@ def process(self, p: StableDiffusionProcessing, enabled: bool = False, b64: str self.set_p_value(p, 'control_net_enabled', cn_num, True) self.set_p_value(p, 'control_net_input_image', cn_num, image) - def postprocess(self, p, processed, enabled: bool = False, b64: str = ''): + def postprocess(self, p, processed, enabled: bool = False, b64: str = '', cn_num: int = 0): if not enabled or b64 is None or len(b64) == 0: return