MediaStreamInput: add "image_batch" option to format input, which downloads a video and outputs as tensor batch (WIP)

This commit is contained in:
Sebastian Monroy
2025-09-01 10:36:20 +01:00
parent f731a55292
commit b091d3057b
+3 -1
View File
@@ -41,7 +41,7 @@ class MediaStreamInput:
return {
"required": {
"input_name": ("STRING", {"default": "default_input", "multiline": False}),
"format": (["auto", "image", "video"],),
"format": (["auto", "image", "video", "image_batch"],),
"presigned_download_url": ("STRING", {
"multiline": True,
"default": "<auto-filled by system>"
@@ -75,6 +75,8 @@ class MediaStreamInput:
return self._process_image(media_bytes)
elif format == 'video':
return self._process_video(media_bytes)
elif format == 'image_batch':
return self._process_video(media_bytes) # Treat as video for processing
else: # format == 'image'
return self._process_image(media_bytes)