From 3c25f5dc0f52788d41b9a44a9e03b845df06b3e8 Mon Sep 17 00:00:00 2001 From: pythongosssss <125205205+pythongosssss@users.noreply.github.com> Date: Wed, 22 Mar 2023 17:32:01 +0000 Subject: [PATCH 1/2] Add content disposition header so images have name --- server.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/server.py b/server.py index 6615a39e4..73429acca 100644 --- a/server.py +++ b/server.py @@ -129,12 +129,12 @@ class PromptServer(): return web.Response(status=403) output_dir = full_output_dir - file = request.rel_url.query["filename"] - file = os.path.basename(file) - file = os.path.join(output_dir, file) + filename = request.rel_url.query["filename"] + filename = os.path.basename(filename) + file = os.path.join(output_dir, filename) if os.path.isfile(file): - return web.FileResponse(file) + return web.FileResponse(file, headers={"Content-Disposition": f"filename=\"{filename}\""}) return web.Response(status=404) From cc127eeabd5cdd42f1a472e43938c6a202e89626 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Thu, 23 Mar 2023 00:40:48 -0400 Subject: [PATCH 2/2] Lower PNG compression for more speed. After some quick tests compress_level=4 seems to give the best compression performance ratio for stable diffusion images. It's 10x faster than max compression for an image size increase of only 2.5%. --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index 530f4cea9..0b8be7659 100644 --- a/nodes.py +++ b/nodes.py @@ -779,7 +779,7 @@ class SaveImage: metadata.add_text(x, json.dumps(extra_pnginfo[x])) file = f"{filename}_{counter:05}_.png" - img.save(os.path.join(full_output_folder, file), pnginfo=metadata, optimize=True) + img.save(os.path.join(full_output_folder, file), pnginfo=metadata, compress_level=4) results.append({ "filename": file, "subfolder": subfolder,