diff --git a/drawing/sdwebui.py b/drawing/sdwebui.py index c79b2c08..788594f5 100644 --- a/drawing/sdwebui.py +++ b/drawing/sdwebui.py @@ -30,8 +30,8 @@ async def text_to_img(self, prompt): resp.raise_for_status() r = resp.json() - for i in r.get('images', []): - return Image(base64=i) + return [Image(base64=i) for i in r.get('images', [])] + async def img_to_img(self, init_images: List[Image], prompt=''): payload = {