Skip to content

Commit 998cc44

Browse files
committed
Add Gemini-3-Flash-preview and others
1 parent 71107a8 commit 998cc44

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

gemini_nodes.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -159,14 +159,15 @@ class SSL_GeminiTextPrompt(cio.ComfyNode):
159159
"gemini-2.0-flash-thinking-exp-1219", "gemini-2.5-pro",
160160
"gemini-2.5-pro-preview-05-06", "gemini-2.5-flash",
161161
"gemini-2.5-flash-preview-09-2025",
162-
"gemini-3-pro-preview"
162+
"gemini-3-pro-preview",
163+
"gemini-3-flash-preview"
163164
]
164-
IMAGE_MODELS = ["gemini-2.5-flash-image-preview", "gemini-3-pro-image-preview"]
165+
IMAGE_MODELS = ["gemini-2.5-flash-image-preview", "gemini-2.5-flash-image", "gemini-3-pro-image-preview", "nano-banana-pro-preview"]
165166
MEDIA_RES_MODELS = [
166167
"gemini-2.0-flash-thinking-exp", "gemini-2.0-flash-thinking-exp-01-21",
167168
"gemini-2.0-flash-thinking-exp-1219", "gemini-2.5-pro",
168169
"gemini-2.5-pro-preview-05-06", "gemini-2.5-flash",
169-
"gemini-3-pro-preview"
170+
"gemini-3-pro-preview", "gemini-3-flash-preview"
170171
]
171172

172173
@classmethod
@@ -179,7 +180,7 @@ def define_schema(cls) -> cio.Schema:
179180
cls.GemConfig.Input("config"),
180181
cio.String.Input("prompt", multiline=True),
181182
cio.String.Input("system_instruction", default="You are a helpful AI assistant.", multiline=True),
182-
cio.Combo.Input("model", options=["learnlm-2.0-flash-experimental", "gemini-exp-1206", "gemini-2.0-flash", "gemini-2.0-flash-lite-001", "gemini-2.0-flash-exp", "gemini-2.0-flash-thinking-exp", "gemini-2.0-flash-thinking-exp-01-21", "gemini-2.0-flash-thinking-exp-1219", "gemini-2.5-pro", "gemini-2.5-pro-preview-05-06", "gemini-2.5-flash", "gemini-2.5-flash-preview-09-2025", "gemini-3-pro-preview", "gemini-2.5-flash-image-preview"], default="gemini-2.0-flash"),
183+
cio.Combo.Input("model", options=["learnlm-2.0-flash-experimental", "gemini-exp-1206", "gemini-2.0-flash", "gemini-2.0-flash-lite-001", "gemini-2.0-flash-exp", "gemini-2.0-flash-thinking-exp", "gemini-2.0-flash-thinking-exp-01-21", "gemini-2.0-flash-thinking-exp-1219", "gemini-2.5-pro", "gemini-2.5-pro-preview-05-06", "gemini-2.5-flash", "gemini-2.5-flash-preview-09-2025", "gemini-2.5-flash-lite-preview-09-2025", "gemini-2.5-flash-lite", "gemini-3-pro-preview", "gemini-2.5-flash-image-preview", "nano-banana-pro-preview", "gemini-3-flash-preview"], default="gemini-2.0-flash"),
183184
cio.Float.Input("temperature", default=1.0, min=0.0, max=1.0, step=0.01),
184185
cio.Float.Input("top_p", default=0.95, min=0.0, max=1.0, step=0.01),
185186
cio.Int.Input("top_k", default=40, min=1, max=100, step=1),

0 commit comments

Comments
 (0)