From 3cf467111151b30a34b21d8d2a9604d9c580775c Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Wed, 30 Apr 2025 17:25:03 -0500 Subject: [PATCH] Proper ray-1-6 support as fix has been applied in backend (#80) --- comfy_api_nodes/apis/luma_api.py | 2 +- comfy_api_nodes/nodes_luma.py | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/comfy_api_nodes/apis/luma_api.py b/comfy_api_nodes/apis/luma_api.py index 037b2415c..632c4ab96 100644 --- a/comfy_api_nodes/apis/luma_api.py +++ b/comfy_api_nodes/apis/luma_api.py @@ -234,7 +234,7 @@ class LumaImageGenerationRequest(BaseModel): class LumaGenerationRequest(BaseModel): prompt: str = Field(..., description='The prompt of the generation') model: LumaVideoModel = Field(LumaVideoModel.ray_2, description='The video model used for the generation') - duration: LumaVideoModelOutputDuration = Field(LumaVideoModelOutputDuration.dur_5s, description='The duration of the generation') + duration: Optional[LumaVideoModelOutputDuration] = Field(None, description='The duration of the generation') aspect_ratio: Optional[LumaAspectRatio] = Field(None, description='The aspect ratio of the generation') resolution: Optional[LumaVideoOutputResolution] = Field(None, description='The resolution of the generation') loop: Optional[bool] = Field(None, description='Whether to loop the video') diff --git a/comfy_api_nodes/nodes_luma.py b/comfy_api_nodes/nodes_luma.py index 32f8cb392..7b95a1f41 100644 --- a/comfy_api_nodes/nodes_luma.py +++ b/comfy_api_nodes/nodes_luma.py @@ -484,6 +484,9 @@ class LumaTextToVideoGenerationNode(ComfyNodeABC): auth_token=None, **kwargs, ): + duration = duration if model != LumaVideoModel.ray_1_6 else None + resolution = resolution if model != LumaVideoModel.ray_1_6 else None + operation = SynchronousOperation( endpoint=ApiEndpoint( path="/proxy/luma/generations", @@ -610,6 +613,8 @@ class LumaImageToVideoGenerationNode(ComfyNodeABC): "At least one of first_image and last_image requires an input." ) keyframes = self._convert_to_keyframes(first_image, last_image, auth_token) + duration = duration if model != LumaVideoModel.ray_1_6 else None + resolution = resolution if model != LumaVideoModel.ray_1_6 else None operation = SynchronousOperation( endpoint=ApiEndpoint(