From 60b459bb4c63402be1a8496ce07ee3fc8f961a6a Mon Sep 17 00:00:00 2001 From: Chenlei Hu Date: Tue, 18 Mar 2025 17:12:02 -0400 Subject: [PATCH] Change SaveWEBM node's output key from images to video --- comfy_extras/nodes_video.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/comfy_extras/nodes_video.py b/comfy_extras/nodes_video.py index 97ca513d8..091080675 100644 --- a/comfy_extras/nodes_video.py +++ b/comfy_extras/nodes_video.py @@ -28,15 +28,12 @@ class SaveWEBM: } RETURN_TYPES = () - FUNCTION = "save_images" - + FUNCTION = "save_video" OUTPUT_NODE = True - - CATEGORY = "image/video" - + CATEGORY = "video" EXPERIMENTAL = True - def save_images(self, images, codec, fps, filename_prefix, crf, prompt=None, extra_pnginfo=None): + def save_video(self, images, codec, fps, filename_prefix, crf, prompt=None, extra_pnginfo=None): 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]) @@ -71,7 +68,7 @@ class SaveWEBM: "type": self.type }] - return {"ui": {"images": results, "animated": (True,)}} # TODO: frontend side + return {"ui": {"video": results}} NODE_CLASS_MAPPINGS = {