File size: 2,125 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 43 44 45 46 47 48 49 50 51 52 |
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)",
} |