@@ -54,10 +54,6 @@ class VeoVideoGenerationNode(ComfyNodeABC):
5454 """
5555 Generates videos from text prompts using Google's Veo API.
5656
57- Supported models:
58- - veo-2.0-generate-001
59- - veo-3.0-generate-preview
60-
6157 This node can create videos from text descriptions and optional image inputs,
6258 with control over parameters like aspect ratio, duration, and more.
6359 """
@@ -134,14 +130,6 @@ def INPUT_TYPES(s):
134130 "default" : None ,
135131 "tooltip" : "Optional reference image to guide video generation" ,
136132 }),
137- "model" : (
138- IO .COMBO ,
139- {
140- "options" : ["veo-2.0-generate-001" , "veo-3.0-generate-preview" ],
141- "default" : "veo-2.0-generate-001" ,
142- "tooltip" : "Model to use for video generation. Defaults to veo 2.0" ,
143- },
144- ),
145133 },
146134 "hidden" : {
147135 "auth_token" : "AUTH_TOKEN_COMFY_ORG" ,
@@ -166,7 +154,6 @@ def generate_video(
166154 person_generation = "ALLOW" ,
167155 seed = 0 ,
168156 image = None ,
169- model = "veo-2.0-generate-001" ,
170157 unique_id : Optional [str ] = None ,
171158 ** kwargs ,
172159 ):
@@ -205,7 +192,7 @@ def generate_video(
205192 # Initial request to start video generation
206193 initial_operation = SynchronousOperation (
207194 endpoint = ApiEndpoint (
208- path = f "/proxy/veo/ { model } /generate" ,
195+ path = "/proxy/veo/generate" ,
209196 method = HttpMethod .POST ,
210197 request_model = Veo2GenVidRequest ,
211198 response_model = Veo2GenVidResponse
@@ -236,7 +223,7 @@ def progress_extractor(response):
236223 # Define the polling operation
237224 poll_operation = PollingOperation (
238225 poll_endpoint = ApiEndpoint (
239- path = f "/proxy/veo/ { model } /poll" ,
226+ path = "/proxy/veo/poll" ,
240227 method = HttpMethod .POST ,
241228 request_model = Veo2GenVidPollRequest ,
242229 response_model = Veo2GenVidPollResponse
@@ -317,5 +304,5 @@ def progress_extractor(response):
317304}
318305
319306NODE_DISPLAY_NAME_MAPPINGS = {
320- "VeoVideoGenerationNode" : "Google Veo Video Generation" ,
307+ "VeoVideoGenerationNode" : "Google Veo2 Video Generation" ,
321308}
0 commit comments