fix: sdwebui send image

lss233 · Apr 17, 2023 · 4ddfcd8 · 4ddfcd8
1 parent 2bf9761
commit 4ddfcd8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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 = {