ComfyUI-HunyuanLoom/__init__.py at main · logtd/ComfyUI-HunyuanLoom · GitHub

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
53
54
55
56
57
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, HYFlowEditGuiderAdvNode, HYFlowEditSamplerNode, HYFlowEditGuiderCFGNode, HYFlowEditGuiderCFGAdvNode

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,
    "HYFlowEditGuiderAdv": HYFlowEditGuiderAdvNode,
    "HYFlowEditGuiderCFG": HYFlowEditGuiderCFGNode,
    "HYFlowEditGuiderCFGAdv": HYFlowEditGuiderCFGAdvNode,
    "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",
    "HYFlowEditGuiderCFG": "HY FlowEdit Guider CFG",
    "HYFlowEditGuiderCFGAdv": "HY FlowEdit Guider CFG Adv.",
    "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)",
}