298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix SaveAnimatedWEBP not working when metadata is disabled.

Browse Source
This commit is contained in:
comfyanonymous 2023-11-23 15:06:35 -05:00
parent 4d2437e681
commit 022033a0e7
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_extras/nodes_images.py
View File

@ -75,7 +75,7 @@ class SaveAnimatedWEBP:
CATEGORY = "_for_testing" CATEGORY = "_for_testing"
def save_images(self, images, fps, filename_prefix, lossless, quality, method, num_frames=0, prompt=None, extra_pnginfo=None): def save_images(self, images, fps, filename_prefix, lossless, quality, method, num_frames=0, prompt=None, extra_pnginfo=None):
method = self.methods.get(method, "aoeu") method = self.methods.get(method)
filename_prefix += self.prefix_append filename_prefix += self.prefix_append
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir, images[0].shape[1], images[0].shape[0]) full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir, images[0].shape[1], images[0].shape[0])
results = list() results = list()
@ -85,9 +85,8 @@ class SaveAnimatedWEBP:
img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8)) img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
pil_images.append(img) pil_images.append(img)
metadata = None metadata = pil_images[0].getexif()
if not args.disable_metadata: if not args.disable_metadata:
metadata = pil_images[0].getexif()
if prompt is not None: if prompt is not None:
metadata[0x0110] = "prompt:{}".format(json.dumps(prompt)) metadata[0x0110] = "prompt:{}".format(json.dumps(prompt))
if extra_pnginfo is not None: if extra_pnginfo is not None: