xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2025-12-10 06:24:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(api-nodes): add Veo3.1 model (#10357)

Browse Source
This commit is contained in:
Alexander Piskun 2025-10-16 01:41:45 +03:00 committed by GitHub
parent f72c6616b2
commit 74b7f0b04b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy_api_nodes/nodes_veo2.py
View File

@ -27,6 +27,13 @@ from comfy_api_nodes.apinode_utils import (
) )
AVERAGE_DURATION_VIDEO_GEN = 32 AVERAGE_DURATION_VIDEO_GEN = 32
MODELS_MAP = {
"veo-2.0-generate-001": "veo-2.0-generate-001",
"veo-3.1-generate": "veo-3.1-generate-preview",
"veo-3.1-fast-generate": "veo-3.1-fast-generate-preview",
"veo-3.0-generate-001": "veo-3.0-generate-001",
"veo-3.0-fast-generate-001": "veo-3.0-fast-generate-001",
}
def convert_image_to_base64(image: torch.Tensor): def convert_image_to_base64(image: torch.Tensor):
if image is None: if image is None:
@ -158,6 +165,7 @@ class VeoVideoGenerationNode(IO.ComfyNode):
model="veo-2.0-generate-001", model="veo-2.0-generate-001",
generate_audio=False, generate_audio=False,
): ):
model = MODELS_MAP[model]
# Prepare the instances for the request # Prepare the instances for the request
instances = [] instances = []
@ -385,7 +393,7 @@ class Veo3VideoGenerationNode(VeoVideoGenerationNode):
), ),
IO.Combo.Input( IO.Combo.Input(
"model", "model",
options=["veo-3.0-generate-001", "veo-3.0-fast-generate-001"], options=list(MODELS_MAP.keys()),
default="veo-3.0-generate-001", default="veo-3.0-generate-001",
tooltip="Veo 3 model to use for video generation", tooltip="Veo 3 model to use for video generation",
optional=True, optional=True,