@@ -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