wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-30 19:07:25 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

[Partner Nodes] fix: removed "beta" models versions from Grok nodes (#14170)

Browse Source
This commit is contained in:
Alexander Piskun 2026-05-29 19:08:43 +03:00 committed by GitHub
parent e7214d78ee
commit ea5b092576
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_api_nodes/nodes_grok.py
View File

@ -58,7 +58,6 @@ class GrokImageNode(IO.ComfyNode):
"grok-imagine-image-quality", "grok-imagine-image-quality",
"grok-imagine-image-pro", "grok-imagine-image-pro",
"grok-imagine-image", "grok-imagine-image",
"grok-imagine-image-beta",
], ],
), ),
IO.String.Input( IO.String.Input(
@ -233,7 +232,6 @@ class GrokImageEditNode(IO.ComfyNode):
"grok-imagine-image-quality", "grok-imagine-image-quality",
"grok-imagine-image-pro", "grok-imagine-image-pro",
"grok-imagine-image", "grok-imagine-image",
"grok-imagine-image-beta",
], ],
), ),
IO.Image.Input("image", display_name="images"), IO.Image.Input("image", display_name="images"),
@ -506,7 +504,7 @@ class GrokVideoNode(IO.ComfyNode):
category="video/partner/Grok", category="video/partner/Grok",
description="Generate video from a prompt or an image", description="Generate video from a prompt or an image",
inputs=[ inputs=[
IO.Combo.Input("model", options=["grok-imagine-video", "grok-imagine-video-beta"]), IO.Combo.Input("model", options=["grok-imagine-video"]),
IO.String.Input( IO.String.Input(
"prompt", "prompt",
multiline=True, multiline=True,
@ -576,8 +574,6 @@ class GrokVideoNode(IO.ComfyNode):
seed: int, seed: int,
image: Input.Image | None = None, image: Input.Image | None = None,
) -> IO.NodeOutput: ) -> IO.NodeOutput:
if model == "grok-imagine-video-beta":
model = "grok-imagine-video"
image_url = None image_url = None
if image is not None: if image is not None:
if get_number_of_images(image) != 1: if get_number_of_images(image) != 1:
@ -618,7 +614,7 @@ class GrokVideoEditNode(IO.ComfyNode):
category="video/partner/Grok", category="video/partner/Grok",
description="Edit an existing video based on a text prompt.", description="Edit an existing video based on a text prompt.",
inputs=[ inputs=[
IO.Combo.Input("model", options=["grok-imagine-video", "grok-imagine-video-beta"]), IO.Combo.Input("model", options=["grok-imagine-video"]),
IO.String.Input( IO.String.Input(
"prompt", "prompt",
multiline=True, multiline=True,