mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-06-17 13:30:16 +08:00
[Partner Nodes] feat: add seed input to Flux Erase node (#14283)
Signed-off-by: bigcat88 <bigcat88@icloud.com>
This commit is contained in:
parent
1f9e7df52a
commit
27b5c423a6
@ -43,6 +43,7 @@ class BFLFluxEraseRequest(BaseModel):
|
|||||||
"white (255) marks areas to remove, black (0) marks areas to preserve.",
|
"white (255) marks areas to remove, black (0) marks areas to preserve.",
|
||||||
)
|
)
|
||||||
dilate_pixels: int = Field(10)
|
dilate_pixels: int = Field(10)
|
||||||
|
seed: int | None = Field(None)
|
||||||
output_format: str = Field("png")
|
output_format: str = Field("png")
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@ -534,6 +534,15 @@ class FluxEraseNode(IO.ComfyNode):
|
|||||||
max=25,
|
max=25,
|
||||||
tooltip="Expands the mask boundaries to ensure clean coverage of the object's edges.",
|
tooltip="Expands the mask boundaries to ensure clean coverage of the object's edges.",
|
||||||
),
|
),
|
||||||
|
IO.Int.Input(
|
||||||
|
"seed",
|
||||||
|
default=0,
|
||||||
|
min=0,
|
||||||
|
max=2147483647,
|
||||||
|
control_after_generate=True,
|
||||||
|
tooltip="The random seed used for creating the noise.",
|
||||||
|
optional=True,
|
||||||
|
),
|
||||||
],
|
],
|
||||||
outputs=[IO.Image.Output()],
|
outputs=[IO.Image.Output()],
|
||||||
hidden=[
|
hidden=[
|
||||||
@ -553,6 +562,7 @@ class FluxEraseNode(IO.ComfyNode):
|
|||||||
image: Input.Image,
|
image: Input.Image,
|
||||||
mask: Input.Image,
|
mask: Input.Image,
|
||||||
dilate_pixels: int = 10,
|
dilate_pixels: int = 10,
|
||||||
|
seed: int = 0,
|
||||||
) -> IO.NodeOutput:
|
) -> IO.NodeOutput:
|
||||||
validate_image_dimensions(image, min_width=256, min_height=256)
|
validate_image_dimensions(image, min_width=256, min_height=256)
|
||||||
mask = resize_mask_to_image(mask, image)
|
mask = resize_mask_to_image(mask, image)
|
||||||
@ -565,6 +575,7 @@ class FluxEraseNode(IO.ComfyNode):
|
|||||||
image=tensor_to_base64_string(image[:, :, :, :3]), # make sure image will have alpha channel removed
|
image=tensor_to_base64_string(image[:, :, :, :3]), # make sure image will have alpha channel removed
|
||||||
mask=mask,
|
mask=mask,
|
||||||
dilate_pixels=dilate_pixels,
|
dilate_pixels=dilate_pixels,
|
||||||
|
seed=seed,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user