File size: 2,113 Bytes
82ea528 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
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',
}
|