wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-10 09:12:31 +08:00
Code Issues Actions 25 Packages Projects Releases Wiki Activity

[Partner Nodes] add grok-imagine-image-quality model (#13725)

Browse Source 
* feat(api-nodes): add grok-imagine-image-quality model

Signed-off-by: bigcat88 <bigcat88@icloud.com>

* fixed price badges

Signed-off-by: bigcat88 <bigcat88@icloud.com>

* fix: adjust price badges

Signed-off-by: bigcat88 <bigcat88@icloud.com>

---------

Signed-off-by: bigcat88 <bigcat88@icloud.com>
Co-authored-by: Jedrzej Kosinski <kosinkadink1@gmail.com>
This commit is contained in:
Alexander Piskun 2026-05-07 09:45:59 +03:00 committed by Jedrzej Kosinski
parent c14164421a
commit 3160eee1af
1 changed files with 27 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
comfy_api_nodes/nodes_grok.py
View File

@ -54,7 +54,12 @@ class GrokImageNode(IO.ComfyNode):
inputs=[ inputs=[
IO.Combo.Input( IO.Combo.Input(
"model", "model",
options=["grok-imagine-image-pro", "grok-imagine-image", "grok-imagine-image-beta"], options=[
"grok-imagine-image-quality",
"grok-imagine-image-pro",
"grok-imagine-image",
"grok-imagine-image-beta",
],
), ),
IO.String.Input( IO.String.Input(
"prompt", "prompt",
@ -111,10 +116,12 @@ class GrokImageNode(IO.ComfyNode):
], ],
is_api_node=True, is_api_node=True,
price_badge=IO.PriceBadge( price_badge=IO.PriceBadge(
depends_on=IO.PriceBadgeDepends(widgets=["model", "number_of_images"]), depends_on=IO.PriceBadgeDepends(widgets=["model", "number_of_images", "resolution"]),
expr=""" expr="""
( (
$rate := $contains(widgets.model, "pro") ? 0.07 : 0.02; $rate := widgets.model = "grok-imagine-image-quality"
? (widgets.resolution = "1k" ? 0.05 : 0.07)
: ($contains(widgets.model, "pro") ? 0.07 : 0.02);
{"type":"usd","usd": $rate * widgets.number_of_images} {"type":"usd","usd": $rate * widgets.number_of_images}
) )
""", """,
@ -167,7 +174,12 @@ class GrokImageEditNode(IO.ComfyNode):
inputs=[ inputs=[
IO.Combo.Input( IO.Combo.Input(
"model", "model",
options=["grok-imagine-image-pro", "grok-imagine-image", "grok-imagine-image-beta"], options=[
"grok-imagine-image-quality",
"grok-imagine-image-pro",
"grok-imagine-image",
"grok-imagine-image-beta",
],
), ),
IO.Image.Input("image", display_name="images"), IO.Image.Input("image", display_name="images"),
IO.String.Input( IO.String.Input(
@ -228,11 +240,19 @@ class GrokImageEditNode(IO.ComfyNode):
], ],
is_api_node=True, is_api_node=True,
price_badge=IO.PriceBadge( price_badge=IO.PriceBadge(
depends_on=IO.PriceBadgeDepends(widgets=["model", "number_of_images"]), depends_on=IO.PriceBadgeDepends(widgets=["model", "number_of_images", "resolution"]),
expr=""" expr="""
( (
$rate := $contains(widgets.model, "pro") ? 0.07 : 0.02; $isQualityModel := widgets.model = "grok-imagine-image-quality";
{"type":"usd","usd": 0.002 + $rate * widgets.number_of_images} $isPro := $contains(widgets.model, "pro");
$rate := $isQualityModel
? (widgets.resolution = "1k" ? 0.05 : 0.07)
: ($isPro ? 0.07 : 0.02);
$base := $isQualityModel ? 0.01 : 0.002;
$output := $rate * widgets.number_of_images;
$isPro
? {"type":"usd","usd": $base + $output}
: {"type":"range_usd","min_usd": $base + $output, "max_usd": 3 * $base + $output}
) )
""", """,
), ),