xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-04 00:27:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add safety filter to Kontext.

Browse Source
This commit is contained in:
Robin Huang 2025-05-27 18:21:03 -07:00
parent 8983af3925
commit e14314f97f
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
comfy_api_nodes/nodes_bfl.py
View File

@ -329,6 +329,15 @@ class FluxKontextProImageNode(ComfyNodeABC):
"tooltip": "The random seed used for creating the noise.",
},
),
"safety_tolerance": (
IO.INT,
{
"default": 2,
"min": 0,
"max": 2,
"tooltip": "Safety tolerance for the image generation process",
},
),
"prompt_upsampling": (
IO.BOOLEAN,
{
@ -375,6 +384,7 @@ class FluxKontextProImageNode(ComfyNodeABC):
input_image: torch.Tensor=None,
seed=0,
prompt_upsampling=False,
safety_tolerance=2,
unique_id: Union[str, None] = None,
**kwargs,
):
@ -382,7 +392,7 @@ class FluxKontextProImageNode(ComfyNodeABC):
validate_string(prompt, strip_whitespace=False)
operation = SynchronousOperation(
endpoint=ApiEndpoint(
path="/proxy/bfl/flux-kontext-pro",
path="/proxy/bfl/flux-kontext-pro/generate",
method=HttpMethod.POST,
request_model=BFLFluxKontextProGenerateRequest,
response_model=BFLFluxProGenerateResponse,
@ -405,6 +415,7 @@ class FluxKontextProImageNode(ComfyNodeABC):
if input_image is None
else convert_image_to_base64(input_image)
),
safety_tolerance=safety_tolerance,
),
auth_kwargs=kwargs,
)