From 32e49fa00f7a2361758f8bb279ce5f6f16ab022b Mon Sep 17 00:00:00 2001 From: silveroxides Date: Tue, 23 Dec 2025 19:49:07 +0100 Subject: [PATCH] safety settings config --- gemini_nodes.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/gemini_nodes.py b/gemini_nodes.py index 54d1e52..dd87c53 100644 --- a/gemini_nodes.py +++ b/gemini_nodes.py @@ -395,6 +395,11 @@ class SSL_GeminiTextPrompt(cio.ComfyNode): types.SafetySetting(category="HARM_CATEGORY_SEXUALLY_EXPLICIT", threshold="BLOCK_NONE"), # type: ignore types.SafetySetting(category="HARM_CATEGORY_DANGEROUS_CONTENT", threshold="BLOCK_NONE"), # type: ignore types.SafetySetting(category="HARM_CATEGORY_CIVIC_INTEGRITY", threshold="BLOCK_NONE"), # type: ignore + types.SafetySetting(category=types.HarmCategory.HARM_CATEGORY_HARASSMENT, threshold=types.HarmBlockThreshold.BLOCK_NONE), # type: ignore + types.SafetySetting(category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH, threshold=types.HarmBlockThreshold.BLOCK_NONE), # type: ignore + types.SafetySetting(category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT, threshold=types.HarmBlockThreshold.BLOCK_NONE), # type: ignore + types.SafetySetting(category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT, threshold=types.HarmBlockThreshold.BLOCK_NONE), # type: ignore + types.SafetySetting(category=types.HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY, threshold=types.HarmBlockThreshold.BLOCK_NONE), # type: ignore ] # Modified: Only trigger image config if include_images is True AND the model is actually an image model