diff options
| author | missionfloyd <missionfloyd@users.noreply.github.com> | 2023-05-26 00:53:33 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-26 00:53:33 +0000 |
| commit | 6645f23c4c715b1bc704c88a499b2f4224d7f1e6 (patch) | |
| tree | 6aeb51e366254fe8993856a3db341690bb39dca5 /modules/progress.py | |
| parent | 43bdaa2f0eda79c685792b06a2bd84c65806a48f (diff) | |
| parent | a6e653be26cc05f4438145fa0082816e9fbbf5fc (diff) | |
| download | stable-diffusion-webui-gfx803-6645f23c4c715b1bc704c88a499b2f4224d7f1e6.tar.gz stable-diffusion-webui-gfx803-6645f23c4c715b1bc704c88a499b2f4224d7f1e6.tar.bz2 stable-diffusion-webui-gfx803-6645f23c4c715b1bc704c88a499b2f4224d7f1e6.zip | |
Merge branch 'dev' into reorder-hotkeys
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
|