Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
26 lines
1.1 KiB
Python
26 lines
1.1 KiB
Python
from . import nag_nodes, tpg_nodes, pladis_nodes
|
|
from .pag_nodes import PerturbedAttention, SlidingWindowGuidanceAdvanced, SmoothedEnergyGuidanceAdvanced
|
|
from .pag_trt_nodes import TRTAttachPag, TRTPerturbedAttention
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"PerturbedAttention": PerturbedAttention,
|
|
"SmoothedEnergyGuidanceAdvanced": SmoothedEnergyGuidanceAdvanced,
|
|
"SlidingWindowGuidanceAdvanced": SlidingWindowGuidanceAdvanced,
|
|
"TRTAttachPag": TRTAttachPag,
|
|
"TRTPerturbedAttention": TRTPerturbedAttention,
|
|
**nag_nodes.NODE_CLASS_MAPPINGS,
|
|
**tpg_nodes.NODE_CLASS_MAPPINGS,
|
|
**pladis_nodes.NODE_CLASS_MAPPINGS,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"PerturbedAttention": "Perturbed-Attention Guidance (Advanced)",
|
|
"SmoothedEnergyGuidanceAdvanced": "Smoothed Energy Guidance (Advanced)",
|
|
"SlidingWindowGuidanceAdvanced": "Sliding Window Guidance (Advanced)",
|
|
"TRTAttachPag": "TensorRT Attach PAG",
|
|
"TRTPerturbedAttention": "TensorRT Perturbed-Attention Guidance",
|
|
**nag_nodes.NODE_DISPLAY_NAME_MAPPINGS,
|
|
**tpg_nodes.NODE_DISPLAY_NAME_MAPPINGS,
|
|
**pladis_nodes.NODE_DISPLAY_NAME_MAPPINGS,
|
|
}
|