From a59f01a27dc0e49acaaece36ca402c1014d70e2a Mon Sep 17 00:00:00 2001 From: KhoaVo Date: Sun, 11 May 2025 20:46:05 -0700 Subject: [PATCH] Support embedding workflow in jpeg It's exactly the same as webp. Tested and verified on local machine. --- WAS_Node_Suite.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/WAS_Node_Suite.py b/WAS_Node_Suite.py index e422df1..275aa71 100644 --- a/WAS_Node_Suite.py +++ b/WAS_Node_Suite.py @@ -7403,7 +7403,7 @@ def was_save_images(self, images, output_path='', filename_prefix="ComfyUI", fil img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8)) # Delegate metadata/pnginfo - if extension == 'webp': + if extension == 'webp' or extension in ["jpg", "jpeg"]: img_exif = img.getexif() if embed_workflow == 'true': workflow_metadata = '' @@ -7442,7 +7442,7 @@ def was_save_images(self, images, output_path='', filename_prefix="ComfyUI", fil output_file = os.path.abspath(os.path.join(output_path, file)) if extension in ["jpg", "jpeg"]: img.save(output_file, - quality=quality, optimize=optimize_image, dpi=(dpi, dpi)) + quality=quality, optimize=optimize_image, dpi=(dpi, dpi), exif=exif_data) elif extension == 'webp': img.save(output_file, quality=quality, lossless=lossless_webp, exif=exif_data)