digital-garbageandClaude Sonnet 5.5
34db2b331d
Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
32 lines
1.7 KiB
Python
32 lines
1.7 KiB
Python
from comfy_api.latest import io
|
|
|
|
from ..._core import log
|
|
from . import erase
|
|
|
|
|
|
class FunUINegativeErase(io.ComfyNode):
|
|
@classmethod
|
|
def define_schema(cls) -> io.Schema:
|
|
return io.Schema(
|
|
node_id="FunUINegativeErase",
|
|
display_name="FunUI Negative Erase",
|
|
category="FunUI/Conditioning",
|
|
description="Give the negative prompt a job at CFG 1: take its direction out of the positive "
|
|
"prompt's words. Strength 0 passes the positive through untouched.",
|
|
inputs=[io.Conditioning.Input("positive"), io.Conditioning.Input("negative"),
|
|
io.Float.Input("strength", default=0.0, min=0.0, max=2.0, step=0.05,
|
|
tooltip="0 = off. How much of the negative prompt is taken out of the positive prompt's words (1 = all of it)."),
|
|
io.Combo.Input("mode", options=list(erase.MODES), default="project",
|
|
tooltip="project: removes what the negative describes from each word. subtract: pushes each word away from the negative."),
|
|
io.Boolean.Input("keep_size", default=True,
|
|
tooltip="Put each word back on its original size, changing its direction only.")],
|
|
outputs=[io.Conditioning.Output(display_name="positive"), io.String.Output(display_name="status")],
|
|
)
|
|
|
|
@classmethod
|
|
def execute(cls, positive, negative, strength: float, mode: str, keep_size: bool = True) -> io.NodeOutput:
|
|
out, note = erase.apply(positive, negative, strength, mode, keep_size)
|
|
if note != "off":
|
|
log.once(f"negative_erase:{note}", log.ALERT, "FunUI Negative Erase", note)
|
|
return io.NodeOutput(out, note)
|