wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-15 07:52:35 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

convert "format" to VideoContainer enum

Browse Source
This commit is contained in:
bigcat88 2025-12-02 15:18:53 +02:00
parent 1e4fb4b131
commit 0ae9c77c82
No known key found for this signature in database
GPG Key ID: 1F0BF0EC3CF22721
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_video.py
View File

@ -88,7 +88,7 @@ class SaveVideo(io.ComfyNode):
) )
@classmethod @classmethod
def execute(cls, video: VideoInput, filename_prefix, format, codec) -> io.NodeOutput: def execute(cls, video: VideoInput, filename_prefix, format: str, codec) -> io.NodeOutput:
width, height = video.get_dimensions() width, height = video.get_dimensions()
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path( full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(
filename_prefix, filename_prefix,
@ -108,7 +108,7 @@ class SaveVideo(io.ComfyNode):
file = f"{filename}_{counter:05}_.{VideoContainer.get_extension(format)}" file = f"{filename}_{counter:05}_.{VideoContainer.get_extension(format)}"
video.save_to( video.save_to(
os.path.join(full_output_folder, file), os.path.join(full_output_folder, file),
format=format, format=VideoContainer(format),
codec=codec, codec=codec,
metadata=saved_metadata metadata=saved_metadata
) )