wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-24 17:32:40 +08:00
Code Issues Actions 32 Packages Projects Releases Wiki Activity

fix: add validation before uploading Assets

Browse Source 
Signed-off-by: bigcat88 <bigcat88@icloud.com>
This commit is contained in:
bigcat88 2026-04-22 10:53:34 +03:00
parent e4afcc0b8b
commit 07a8ffbe51
No known key found for this signature in database
GPG Key ID: 1F0BF0EC3CF22721
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
comfy_api_nodes/nodes_bytedance.py
View File

@ -52,6 +52,8 @@ from comfy_api_nodes.util import (
validate_image_aspect_ratio, validate_image_aspect_ratio,
validate_image_dimensions, validate_image_dimensions,
validate_string, validate_string,
validate_video_dimensions,
validate_video_duration,
) )
from server import PromptServer from server import PromptServer
@ -1904,6 +1906,8 @@ class ByteDanceCreateImageAsset(IO.ComfyNode):
) -> IO.NodeOutput: ) -> IO.NodeOutput:
# if len(name) > 64: # if len(name) > 64:
# raise ValueError("Name of asset can not be greater then 64 symbols") # raise ValueError("Name of asset can not be greater then 64 symbols")
validate_image_dimensions(image, min_width=300, max_width=6000, min_height=300, max_height=6000)
validate_image_aspect_ratio(image, min_ratio=(0.4, 1), max_ratio=(2.5, 1))
resolved_group = await _resolve_group_id(cls, group_id) resolved_group = await _resolve_group_id(cls, group_id)
asset_id = await _create_seedance_asset( asset_id = await _create_seedance_asset(
cls, cls,
@ -1965,6 +1969,24 @@ class ByteDanceCreateVideoAsset(IO.ComfyNode):
) -> IO.NodeOutput: ) -> IO.NodeOutput:
# if len(name) > 64: # if len(name) > 64:
# raise ValueError("Name of asset can not be greater then 64 symbols") # raise ValueError("Name of asset can not be greater then 64 symbols")
validate_video_duration(video, min_duration=2, max_duration=15)
validate_video_dimensions(video, min_width=300, max_width=6000, min_height=300, max_height=6000)
w, h = video.get_dimensions()
if h > 0:
ratio = w / h
if not (0.4 <= ratio <= 2.5):
raise ValueError(f"Asset video aspect ratio (W/H) must be in [0.4, 2.5], got {ratio:.3f} ({w}x{h}).")
pixels = w * h
if not (409_600 <= pixels <= 927_408):
raise ValueError(
f"Asset video total pixels (W×H) must be in [409600, 927408], " f"got {pixels:,} ({w}x{h})."
)
fps = float(video.get_frame_rate())
if not (24 <= fps <= 60):
raise ValueError(f"Asset video FPS must be in [24, 60], got {fps:.2f}.")
resolved_group = await _resolve_group_id(cls, group_id) resolved_group = await _resolve_group_id(cls, group_id)
asset_id = await _create_seedance_asset( asset_id = await _create_seedance_asset(
cls, cls,