custom_nodes/ComfyUI-LTXTricks/__init__.py

from .nodes.modify_ltx_model_node import ModifyLTXModelNode
from .nodes.latent_guide_node import AddLatentGuideNode
from .nodes.ltx_inverse_model_pred_nodes import LTXForwardModelSamplingPredNode, LTXReverseModelSamplingPredNode
from .nodes.rectified_sampler_nodes import LTXRFForwardODESamplerNode, LTXRFReverseODESamplerNode
from .nodes.attn_bank_nodes import LTXPrepareAttnInjectionsNode, LTXAttentionBankNode, LTXAttentioOverrideNode
from .nodes.ltx_pag_node import LTXPerturbedAttentionNode
from .nodes.attn_override_node import LTXAttnOverrideNode
from .nodes.ltx_flowedit_nodes import LTXFlowEditCFGGuiderNode, LTXFlowEditSamplerNode


NODE_CLASS_MAPPINGS = {
    'ModifyLTXModel': ModifyLTXModelNode,
    'AddLatentGuide': AddLatentGuideNode,
    'LTXForwardModelSamplingPred': LTXForwardModelSamplingPredNode,
    'LTXReverseModelSamplingPred': LTXReverseModelSamplingPredNode,
    'LTXRFForwardODESampler': LTXRFForwardODESamplerNode,
    'LTXRFReverseODESampler': LTXRFReverseODESamplerNode,
    'LTXAttentionBank': LTXAttentionBankNode,
    'LTXPrepareAttnInjections': LTXPrepareAttnInjectionsNode,
    'LTXAttentioOverride': LTXAttentioOverrideNode,
    'LTXPerturbedAttention': LTXPerturbedAttentionNode,
    'LTXAttnOverride': LTXAttnOverrideNode,
    'LTXFlowEditCFGGuider': LTXFlowEditCFGGuiderNode,
    'LTXFlowEditSampler': LTXFlowEditSamplerNode,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    'ModifyLTXModel': 'Modify LTX Model',
    'AddLatentGuide': 'Add LTX Latent Guide',
    'LTXForwardModelSamplingPred': 'LTX Forward Model Pred',
    'LTXReverseModelSamplingPred': 'LTX Reverse Model Pred',
    'LTXRFForwardODESampler': 'LTX Rf-Inv Forward Sampler',
    'LTXRFReverseODESampler': 'LTX Rf-Inv Reverse Sampler',
    'LTXAttentionBank': 'LTX Attention Bank',
    'LTXPrepareAttnInjections': 'LTX Prepare Attn Injection',
    'LTXAttentioOverride': 'LTX Attn Block Override',
    'LTXPerturbedAttention': 'LTX Apply Perturbed Attention',
    'LTXAttnOverride': 'LTX Attention Override',
    'LTXFlowEditCFGGuider': 'LTX Flow Edit CFG Guider',
    'LTXFlowEditSampler': 'LTX Flow Edit Sampler',
}