Claude
6fcd3b8a86
Rebase the DSL text encoder on comfy.sd1_clip.SDClipModel and comfy.clip_model.CLIPTextModel_, replacing the old HuggingFace transformers CLIPTextModel/CLIPTextTransformer/CLIPTextEmbeddings subclasses (which are no longer how ComfyUI implements CLIP). Structure: - Merge lib/action/ into lib/actions/; drop lib/fun_clip_stuff.py and the bundled clip_config*.json (comfy ships its own) - Convert all custom_nodes.KepPromptLang.* absolute imports to relative imports so installs via Manager work regardless of install directory name Encoder: - PromptLangSDClipModel overrides encode_token_weights to walk the segment/action tree, assemble [B, seq, hidden] embeds, and call self.transformer(None, mask, embeds=..., num_tokens=...) - posScale / postPos are supported without patching the transformer by pre-baking the delta (modified - default) into embeds, so the transformer's inline add yields the modified position embedding - Drop the unused empty-baseline batch that was prepended and then sliced off; halves the per-encode forward pass for single prompts Actions: - Fix copy-pasted broken __repr__ / depth_repr across sum/diff/avg/ slerp that referenced fields that didn't exist - Fix NameError in AverageAction._validate_args (start_arg_token_length) - Fix class-level mutable state in RandAction - Unify _parse_scalar / _parse_scalar_weight / _parse_int into a single parse_numeric_arg helper in action_utils.py - Share add_with_broadcast between SumAction and DiffAction - Convert PostModifiers from TypedDict to dataclass (attribute access catches typos that .get() on string keys hides) Drop the two _exp-pooler / _exp-pooledAvg actions: they recursively invoked the HF CLIPTextTransformer and would need a rework to fit the current CLIPTextModel_ interface. They were experimental and not documented as stable. BuildGif node: - Collapse the 10-positional-arg _save_* helpers onto a small _SaveContext dataclass - Stop mutating the input arg semantics (split_every_val was reassigned to len(images) when -1) Tests: - Add pytest suite covering the parser (grammar, nesting, errors) and every action (embedding math, shape, modifier payloads) - conftest stubs ComfyUI at collection time so tests don't need a real ComfyUI install Drop the broken test_files/ scripts (CI helpers, not a test suite) and regenerate the README from tools/build_docs.py.
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from typing import List
|
|
|
|
from torch import Tensor
|
|
from torch.nn import Embedding
|
|
|
|
from .action_utils import concat_embeddings, get_total_length, parse_numeric_arg
|
|
from .base import MultiArgAction
|
|
from .types import SegOrAction
|
|
|
|
|
|
class MultiplyAction(MultiArgAction):
|
|
grammar = 'mult(" arg+ ")"'
|
|
|
|
display_name = "Multiply"
|
|
action_name = "mult"
|
|
description = "Multiplies the provided segments or actions by the multiplier."
|
|
usage_examples = [
|
|
"mult(The cat is|2.5)",
|
|
"mult(Cat|-1)",
|
|
]
|
|
|
|
def __init__(self, args: List[List[SegOrAction]]) -> None:
|
|
super().__init__(args)
|
|
if len(args) != 2:
|
|
raise ValueError("Multiply action expects exactly two arguments")
|
|
|
|
self.target_arg = args[0]
|
|
self.parsed_multiplier = parse_numeric_arg(
|
|
args[1], action_name="Multiply", role="multiplier", cast=float
|
|
)
|
|
|
|
def token_length(self) -> int:
|
|
return get_total_length(self.target_arg)
|
|
|
|
def get_result(self, embedding_module: Embedding) -> Tensor:
|
|
return concat_embeddings(self.target_arg, embedding_module) * self.parsed_multiplier
|