wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-27 19:02:31 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

nodes_video: save stream: Make audio optional

Browse Source
This commit is contained in:
Rattus 2026-04-14 11:12:47 +10:00
parent b23f1f456f
commit 0c70446c9b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_video.py
View File

@ -278,11 +278,11 @@ class SavePassthroughVideoStream(io.ComfyNode):
has_intermediate_output=True, has_intermediate_output=True,
inputs=[ inputs=[
io.ImageStream.Input("stream", tooltip="The image stream to save."), io.ImageStream.Input("stream", tooltip="The image stream to save."),
io.Audio.Input("audio", optional=True, tooltip="The audio to add to the video."),
io.Float.Input("fps", default=30.0, min=1.0, max=120.0, step=1.0), io.Float.Input("fps", default=30.0, min=1.0, max=120.0, step=1.0),
io.String.Input("filename_prefix", default="video/ComfyUI", tooltip="The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes."), io.String.Input("filename_prefix", default="video/ComfyUI", tooltip="The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes."),
io.Combo.Input("format", options=Types.VideoContainer.as_input(), default="auto", tooltip="The format to save the video as."), io.Combo.Input("format", options=Types.VideoContainer.as_input(), default="auto", tooltip="The format to save the video as."),
io.Combo.Input("codec", options=Types.VideoCodec.as_input(), default="auto", tooltip="The codec to use for the video."), io.Combo.Input("codec", options=Types.VideoCodec.as_input(), default="auto", tooltip="The codec to use for the video."),
io.Audio.Input("audio", optional=True, tooltip="The audio to add to the video."),
], ],
outputs=[ outputs=[
io.ImageStream.Output(display_name="passthrough"), io.ImageStream.Output(display_name="passthrough"),
@ -294,11 +294,11 @@ class SavePassthroughVideoStream(io.ComfyNode):
def execute( def execute(
cls, cls,
stream: Input.ImageStream, stream: Input.ImageStream,
audio: Optional[Input.Audio],
fps: float, fps: float,
filename_prefix, filename_prefix,
format: str, format: str,
codec, codec,
audio: Optional[Input.Audio] = None,
) -> io.NodeOutput: ) -> io.NodeOutput:
return io.NodeOutput(_build_saved_stream(cls.hidden, stream, audio, fps, filename_prefix, format, codec)) return io.NodeOutput(_build_saved_stream(cls.hidden, stream, audio, fps, filename_prefix, format, codec))
@ -316,12 +316,12 @@ class SaveVideoStream(io.ComfyNode):
is_output_node=True, is_output_node=True,
inputs=[ inputs=[
io.ImageStream.Input("stream", tooltip="The image stream to save."), io.ImageStream.Input("stream", tooltip="The image stream to save."),
io.Audio.Input("audio", optional=True, tooltip="The audio to add to the video."),
io.Float.Input("fps", default=30.0, min=1.0, max=120.0, step=1.0), io.Float.Input("fps", default=30.0, min=1.0, max=120.0, step=1.0),
io.Int.Input("chunk_size", default=8, min=1, max=4096), io.Int.Input("chunk_size", default=8, min=1, max=4096),
io.String.Input("filename_prefix", default="video/ComfyUI", tooltip="The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes."), io.String.Input("filename_prefix", default="video/ComfyUI", tooltip="The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes."),
io.Combo.Input("format", options=Types.VideoContainer.as_input(), default="auto", tooltip="The format to save the video as."), io.Combo.Input("format", options=Types.VideoContainer.as_input(), default="auto", tooltip="The format to save the video as."),
io.Combo.Input("codec", options=Types.VideoCodec.as_input(), default="auto", tooltip="The codec to use for the video."), io.Combo.Input("codec", options=Types.VideoCodec.as_input(), default="auto", tooltip="The codec to use for the video."),
io.Audio.Input("audio", optional=True, tooltip="The audio to add to the video."),
], ],
outputs=[], outputs=[],
hidden=[io.Hidden.prompt, io.Hidden.extra_pnginfo, io.Hidden.unique_id, io.Hidden.dynprompt], hidden=[io.Hidden.prompt, io.Hidden.extra_pnginfo, io.Hidden.unique_id, io.Hidden.dynprompt],
@ -331,12 +331,12 @@ class SaveVideoStream(io.ComfyNode):
def execute( def execute(
cls, cls,
stream: Input.ImageStream, stream: Input.ImageStream,
audio: Optional[Input.Audio],
fps: float, fps: float,
chunk_size: int, chunk_size: int,
filename_prefix, filename_prefix,
format: str, format: str,
codec, codec,
audio: Optional[Input.Audio] = None,
) -> io.NodeOutput: ) -> io.NodeOutput:
saved_stream = _build_saved_stream( saved_stream = _build_saved_stream(
cls.hidden, cls.hidden,