diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-27 17:13:26 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-27 17:13:26 +0000 |
| commit | 50906bf78bce9cdb872a56d498000316ec2dfdcd (patch) | |
| tree | 5811cd211f383f759120c80ed266ed0b1befbaf0 /modules/progress.py | |
| parent | 89f9faa63388756314e8a1d96cf86bf5e0663045 (diff) | |
| parent | b186045fee0d384addcdc2a759fd33dba51b070e (diff) | |
| download | stable-diffusion-webui-gfx803-50906bf78bce9cdb872a56d498000316ec2dfdcd.tar.gz stable-diffusion-webui-gfx803-50906bf78bce9cdb872a56d498000316ec2dfdcd.tar.bz2 stable-diffusion-webui-gfx803-50906bf78bce9cdb872a56d498000316ec2dfdcd.zip | |
Merge branch 'release_candidate'
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
|