mcDandy
f6413f4202
Improved error messages in AudioMathNode and refactored ConditioningMathNode to simplify input handling and output structure. Updated README to clarify blur and edge function behaviors. Removed redundant 'category' field from deprecated node schemas to hide them.
59 lines
2.5 KiB
Python
59 lines
2.5 KiB
Python
from comfy_api.latest import io
|
|
from ..modelLikeCommon import calculate_patches
|
|
from inspect import cleandoc
|
|
from ..helper_functions import commonLazy
|
|
|
|
|
|
class CLIPMathNodeOLD(io.ComfyNode):
|
|
"""
|
|
This node enables the use of math expressions on CLIP weights.
|
|
"""
|
|
|
|
@classmethod
|
|
def define_schema(cls) -> io.Schema:
|
|
return io.Schema(
|
|
node_id="mrmth_CLIPMathNode",
|
|
display_name="CLIP Math",
|
|
is_deprecated=True,
|
|
inputs=[
|
|
io.Clip.Input(id="a", tooltip="Main CLIP (base)"),
|
|
io.Clip.Input(id="b", optional=True, lazy=True, tooltip="Optional 2nd CLIP"),
|
|
io.Clip.Input(id="c", optional=True, lazy=True, tooltip="Optional 3rd CLIP"),
|
|
io.Clip.Input(id="d", optional=True, lazy=True, tooltip="Optional 4th CLIP"),
|
|
io.Float.Input(id="w", default=0.0, optional=True, lazy=True, force_input=True),
|
|
io.Float.Input(id="x", default=0.0, optional=True, lazy=True, force_input=True),
|
|
io.Float.Input(id="y", default=0.0, optional=True, lazy=True, force_input=True),
|
|
io.Float.Input(id="z", default=0.0, optional=True, lazy=True, force_input=True),
|
|
io.String.Input(id="Model", default="a*(1-w)+b*w", tooltip="Expression to apply on weights"),
|
|
io.Combo.Input(
|
|
id="length_mismatch",
|
|
options=["error", "passthrough", "pad"],
|
|
default="error",
|
|
tooltip="How to handle mismatched layer counts. For models, this usually defaults to broadcast (zero for missing layers)."
|
|
)
|
|
],
|
|
outputs=[
|
|
io.Clip.Output(),
|
|
],
|
|
)
|
|
|
|
tooltip = cleandoc(__doc__)
|
|
|
|
@classmethod
|
|
def check_lazy_status(cls, Model, a, b=[], c=[], d=[], w=0, x=0, y=0, z=0, length_mismatch="tile"):
|
|
return commonLazy(Model, a, b, c, d, w, x, y, z)
|
|
|
|
@classmethod
|
|
def execute(cls, Model, a, b=None, c=None, d=None, w=0.0, x=0.0, y=0.0, z=0.0, length_mismatch="tile") -> io.NodeOutput:
|
|
patcher_a = a.patcher
|
|
patcher_b = b.patcher if b else None
|
|
patcher_c = c.patcher if c else None
|
|
patcher_d = d.patcher if d else None
|
|
|
|
patches = calculate_patches(Model, patcher_a, patcher_b, patcher_c, patcher_d, w, x, y, z)
|
|
|
|
out_clip = a.clone()
|
|
if patches:
|
|
out_clip.add_patches(patches, 1.0, 1.0)
|
|
return (out_clip,)
|