nodes/Aryan185-ComfyUI-ExternalAPI-Helpers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added node for mask segmentation with Gemini

Browse Source
This commit is contained in:
Aryan185
2025-08-29 06:35:03 +00:00
parent 584d98a785
commit fb237a1e02
3 changed files with 192 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
gemini_node.py
+4 -1
View File
@@ -44,7 +44,7 @@ class GeminiChatNode:
# Initialize client and build parts
client = genai.Client(api_key=key)
client = genai.Client(api_key=key, http_options=types.HttpOptions(retry_options=types.HttpRetryOptions(attempts=3, jitter=10)))
parts = [types.Part.from_text(text=prompt)]
# Handle image input
@@ -62,14 +62,17 @@ class GeminiChatNode:
model_lower = model.lower()
if "gemini-2.0" in model_lower:
print("Gemini-2.0 models do not support thinking - disabling thinking config")
final_thinking_budget = None
elif not thinking:
final_thinking_budget = 0
if "gemini-2.5-pro" in model_lower:
print("Gemini-2.5-Pro cannot have thinking turned off - defaulting thinking budget to -1")
final_thinking_budget = -1
else:
final_thinking_budget = thinking_budget
if "gemini-2.5-pro" in model_lower and final_thinking_budget == 0:
print("Gemini-2.5-Pro cannot have thinking turned off - defaulting thinking budget to -1")
final_thinking_budget = -1
config = types.GenerateContentConfig(