Skip to content

Commit cc6aa1d

Browse files
committed
controlnet: use alwayson_scripts
1 parent 264554b commit cc6aa1d

File tree

3 files changed

+13
-6
lines changed

3 files changed

+13
-6
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.0",
14+
version="0.6.1",
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.0"
3+
__version__ = "0.6.1"
44

55
__all__ = [
66
"__version__",

webuiapi/webuiapi.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,7 @@ def txt2img(self,
203203
alwayson_scripts={},
204204
controlnet_units: List[ControlNetUnit] = [],
205205
sampler_index=None, # deprecated: use sampler_name
206+
use_deprecated_controlnet=False,
206207
):
207208
if sampler_index is None:
208209
sampler_index = self.default_sampler
@@ -255,12 +256,18 @@ def txt2img(self,
255256
"save_images": save_images,
256257
"alwayson_scripts": alwayson_scripts,
257258
}
258-
if controlnet_units and len(controlnet_units)>0:
259+
260+
if use_deprecated_controlnet and controlnet_units and len(controlnet_units)>0:
259261
payload["controlnet_units"] = [x.to_dict() for x in controlnet_units]
260262
return self.custom_post('controlnet/txt2img', payload=payload)
261-
else:
262-
response = self.session.post(url=f'{self.baseurl}/txt2img', json=payload)
263-
return self._to_api_result(response)
263+
264+
if controlnet_units and len(controlnet_units)>0:
265+
payload["alwayson_scripts"]["ControlNet"] = {
266+
"args": [x.to_dict() for x in controlnet_units]
267+
}
268+
269+
response = self.session.post(url=f'{self.baseurl}/txt2img', json=payload)
270+
return self._to_api_result(response)
264271

265272
def img2img(self,
266273
images=[], # list of PIL Image

0 commit comments

Comments
 (0)