diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-05-18 07:12:17 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-18 07:12:17 +0000 |
| commit | 182330ae40c93b761fced8d03bbdd8523f1739d0 (patch) | |
| tree | c95f2e531cc34475eb39266add6d12a2a5e1e2b3 /modules/progress.py | |
| parent | 0d31f20cbd556ea4ba3d8ad9254bcce71c32088c (diff) | |
| parent | 983f2c494ad8fed2f08193681ba0bf5dda01555a (diff) | |
| download | stable-diffusion-webui-gfx803-182330ae40c93b761fced8d03bbdd8523f1739d0.tar.gz stable-diffusion-webui-gfx803-182330ae40c93b761fced8d03bbdd8523f1739d0.tar.bz2 stable-diffusion-webui-gfx803-182330ae40c93b761fced8d03bbdd8523f1739d0.zip | |
Merge branch 'dev' into ngrok-py
Diffstat (limited to 'modules/progress.py')
| -rw-r--r-- | modules/progress.py | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/modules/progress.py b/modules/progress.py index 948e6f00..f405f07f 100644 --- a/modules/progress.py +++ b/modules/progress.py @@ -95,9 +95,20 @@ def progressapi(req: ProgressRequest): image = shared.state.current_image
if image is not None:
buffered = io.BytesIO()
- image.save(buffered, format="png")
+
+ if opts.live_previews_image_format == "png":
+ # using optimize for large images takes an enormous amount of time
+ if max(*image.size) <= 256:
+ save_kwargs = {"optimize": True}
+ else:
+ save_kwargs = {"optimize": False, "compress_level": 1}
+
+ 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/png;base64,{base64_image}"
+ 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
|