custom_nodes/ComfyUI-HunyuanLoom/__init__.py

from .nodes.modify_hy_model_node import ConfigureModifiedHYNode
from .nodes.hy_model_pred_nodes import HYInverseModelSamplingPredNode, HYReverseModelSamplingPredNode
from .nodes.rectified_sampler_nodes import HYForwardODESamplerNode, HYReverseODESamplerNode
from .nodes.flowedit_nodes import HYFlowEditGuiderNode, HYFlowEditSamplerNode

from .nodes.hy_regional_cond_nodes import HYApplyRegionalCondsNode, HYCreateRegionalCondNode
from .nodes.hy_attn_override_node import HYAttnOverrideNode

from .nodes.hy_feta_enhance_node import HYFetaEnhanceNode

from .nodes.wrapper_flow_edit_nodes import HyVideoFlowEditSamplerNode


NODE_CLASS_MAPPINGS = {
    "ConfigureModifiedHY": ConfigureModifiedHYNode,
    # RF-Inversion
    "HYInverseModelSamplingPred": HYInverseModelSamplingPredNode,
    "HYReverseModelSamplingPred": HYReverseModelSamplingPredNode,
    "HYForwardODESampler": HYForwardODESamplerNode,
    "HYReverseODESampler": HYReverseODESamplerNode,
    # FlowEdit
    "HYFlowEditGuider": HYFlowEditGuiderNode,
    "HYFlowEditSampler": HYFlowEditSamplerNode,
    # Regional
    "HYApplyRegionalConds": HYApplyRegionalCondsNode,
    "HYCreateRegionalCond": HYCreateRegionalCondNode,
    "HYAttnOverride": HYAttnOverrideNode,
    # Enhance
    "HYFetaEnhance": HYFetaEnhanceNode,
    # Wrapper
    "HyVideoFlowEditSamplerWrapper": HyVideoFlowEditSamplerNode,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "ConfigureModifiedHY": "Modify Hunyuan Model",
    # RF-Inversion
    "HYInverseModelSamplingPred": "HY Inverse Model Pred",
    "HYReverseModelSamplingPred": "HY Reverse Model Pred",
    "HYForwardODESampler": "HY RF-Inv Forward Sampler",
    "HYReverseODESampler": "HY RF-Inv Reverse Sampler",
    # FlowEdit
    "HYFlowEditGuider": "HY FlowEdit Guider",
    "HYFlowEditSampler": "HY FlowEdit Sampler",
    # Regional
    "HYApplyRegionalConds": "HY Apply Regional Conds",
    "HYCreateRegionalCond": "HY Create Regional Cond",
    "HYAttnOverride": "HY Attention Override",
    # Enhance
    "HYFetaEnhance": "HY Feta Enhance",
    # Wrapper
    "HyVideoFlowEditSamplerWrapper": "HunyuanVideo Flow Edit Sampler (Wrapper)",
}