diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-05-14 08:46:27 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-05-14 08:46:27 +0000 |
commit | 80adb6979d46bbb832254004cac4f4f9bec9efb3 (patch) | |
tree | e206ee60f9be21e9e20d483213b7d0a610d2bdbd /modules/progress.py | |
parent | 1dcd6723242c3d691610f9ed937951baea49c2d1 (diff) | |
parent | 3ddc76342298ad0b2d14cb571ceb48c0b0c4176d (diff) | |
download | stable-diffusion-webui-gfx803-80adb6979d46bbb832254004cac4f4f9bec9efb3.tar.gz stable-diffusion-webui-gfx803-80adb6979d46bbb832254004cac4f4f9bec9efb3.tar.bz2 stable-diffusion-webui-gfx803-80adb6979d46bbb832254004cac4f4f9bec9efb3.zip |
Merge branch 'dev' into find_vae
Diffstat (limited to 'modules/progress.py')
-rw-r--r-- | modules/progress.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/modules/progress.py b/modules/progress.py index 5655346b..269863c9 100644 --- a/modules/progress.py +++ b/modules/progress.py @@ -95,8 +95,16 @@ def progressapi(req: ProgressRequest): image = shared.state.current_image
if image is not None:
buffered = io.BytesIO()
- image.save(buffered, format="png")
- live_preview = 'data:image/png;base64,' + base64.b64encode(buffered.getvalue()).decode("ascii")
+
+ if opts.live_previews_image_format == "png":
+ # using optimize for large images takes an enormous amount of time
+ save_kwargs = {"optimize": max(*image.size) > 256}
+ else:
+ save_kwargs = {}
+
+ image.save(buffered, format=opts.live_previews_image_format, **save_kwargs)
+ base64_image = base64.b64encode(buffered.getvalue()).decode('ascii')
+ live_preview = f"data:image/{opts.live_previews_image_format};base64,{base64_image}"
id_live_preview = shared.state.id_live_preview
else:
live_preview = None
|