Setting generation properties

2023-02-26 22:16:05 +02:00
parent 8f0cd0ac05
commit 3c3d644137
10 changed files with 286 additions and 56 deletions
--- a/bot/modules/api/txt2img.py
+++ b/bot/modules/api/txt2img.py
@@ -22,8 +22,10 @@ async def txt2img(prompt: Prompt, ignore_exceptions: bool = False) -> list[bytes
                    "sampler_index": prompt.sampler
                }
            )
-            if r.status != 200:
+            if r.status != 200 and ignore_exceptions:
                return None
+            elif r.status != 200:
+                raise ValueError((await r.json())['detail'])
            return [base64.b64decode((await r.json())["images"][0]),
                    json.loads((await r.json())["info"])]
    except Exception as e: