298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-13 15:03:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change SaveWEBM node's output key from images to video

Browse Source
This commit is contained in:
Chenlei Hu 2025-03-18 17:12:02 -04:00
parent 3b19fc76e3
commit 60b459bb4c
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
comfy_extras/nodes_video.py
View File

@ -28,15 +28,12 @@ class SaveWEBM:
} }
RETURN_TYPES = () RETURN_TYPES = ()
FUNCTION = "save_images" FUNCTION = "save_video"
OUTPUT_NODE = True OUTPUT_NODE = True
CATEGORY = "video"
CATEGORY = "image/video"
EXPERIMENTAL = True 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 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])
@ -71,7 +68,7 @@ class SaveWEBM:
"type": self.type "type": self.type
}] }]
return {"ui": {"images": results, "animated": (True,)}} # TODO: frontend side return {"ui": {"video": results}}
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {