Skip to content

Commit 619179f

Browse files
committed
controlnet: use alwayson_scripts (img2img)
1 parent cc6aa1d commit 619179f

File tree

3 files changed

+12
-7
lines changed

3 files changed

+12
-7
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
setup(
1313
name="webuiapi",
14-
version="0.6.1",
14+
version="0.6.2",
1515
description="Python API client for AUTOMATIC1111/stable-diffusion-webui",
1616
url="https://github.com/mix1009/sdwebuiapi",
1717
author="ChunKoo Park",

webuiapi/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from .webuiapi import WebUIApi, WebUIApiResult, Upscaler, HiResUpscaler, b64_img, raw_b64_img, ModelKeywordResult, ModelKeywordInterface, InstructPix2PixInterface, ControlNetInterface, ControlNetUnit
22

3-
__version__ = "0.6.1"
3+
__version__ = "0.6.2"
44

55
__all__ = [
66
"__version__",

webuiapi/webuiapi.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,7 @@ def img2img(self,
315315
save_images=False,
316316
alwayson_scripts={},
317317
controlnet_units: List[ControlNetUnit] = [],
318+
use_deprecated_controlnet = False,
318319
):
319320
if sampler_name is None:
320321
sampler_name = self.default_sampler
@@ -372,13 +373,17 @@ def img2img(self,
372373
}
373374
if mask_image is not None:
374375
payload['mask'] = b64_img(mask_image)
375-
376-
if controlnet_units and len(controlnet_units)>0:
376+
377+
if use_deprecated_controlnet and controlnet_units and len(controlnet_units)>0:
377378
payload["controlnet_units"] = [x.to_dict() for x in controlnet_units]
378379
return self.custom_post('controlnet/img2img', payload=payload)
379-
else:
380-
response = self.session.post(url=f'{self.baseurl}/img2img', json=payload)
381-
return self._to_api_result(response)
380+
381+
if controlnet_units and len(controlnet_units)>0:
382+
payload["alwayson_scripts"]["ControlNet"] = {
383+
"args": [x.to_dict() for x in controlnet_units]
384+
}
385+
response = self.session.post(url=f'{self.baseurl}/img2img', json=payload)
386+
return self._to_api_result(response)
382387

383388
def extra_single_image(self,
384389
image, # PIL Image

0 commit comments

Comments
 (0)