xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-16 01:36:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
bilt 2025-07-31 19:47:01 +08:00
parent be4b655f01
commit fee1fd8963
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
comfy_api/input_impl/video_types.py
View File

@ -258,11 +258,24 @@ class VideoFromComponents(VideoInput):
codec: VideoCodec = VideoCodec.AUTO, codec: VideoCodec = VideoCodec.AUTO,
metadata: Optional[dict] = None metadata: Optional[dict] = None
): ):
if format != VideoContainer.AUTO and format != VideoContainer.MP4: if format == VideoContainer.AUTO:
format = VideoContainer.MP4
if codec == VideoCodec.AUTO:
codec = VideoCodec.H264
if format != VideoContainer.MP4:
raise ValueError("Only MP4 format is supported for now") raise ValueError("Only MP4 format is supported for now")
if codec != VideoCodec.AUTO and codec != VideoCodec.H264: if codec != VideoCodec.H264:
raise ValueError("Only H264 codec is supported for now") raise ValueError("Only H264 codec is supported for now")
with av.open(path, mode='w', options={'movflags': 'use_metadata_tags'}) as output:
open_kwargs = {
'mode': 'w',
'options': {'movflags': 'use_metadata_tags'}
}
if isinstance(path, io.BytesIO):
open_kwargs['format'] = format.lower()
with av.open(path, **open_kwargs) as output:
# Add metadata before writing any streams # Add metadata before writing any streams
if metadata is not None: if metadata is not None:
for key, value in metadata.items(): for key, value in metadata.items():