-
Notifications
You must be signed in to change notification settings - Fork 32
/
__init__.py
90 lines (86 loc) · 3.81 KB
/
__init__.py
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
## Inversion
from .nodes.flip_sigmas_node import InFluxFlipSigmasNode
from .nodes.influx_model_pred_node import InFluxModelSamplingPredNode, OutFluxModelSamplingPredNode
from .nodes.flux_deguidance_node import FluxDeGuidance
from .nodes.inverse_sampler_node import FluxInverseSamplerNode
from .nodes.configure_modified_flux_node import ConfigureModifiedFluxNode
from .nodes.mix_noise_node import FluxNoiseMixerNode
from .nodes.rectified_sampler_nodes import FluxForwardODESamplerNode, FluxReverseODESamplerNode
from .nodes.rf_edit_sampler_nodes import FlowEditForwardSamplerNode, FlowEditReverseSamplerNode, PrepareAttnBankNode
## Enhance
from .nodes.apply_seg_node import SEGAttentionNode
from .nodes.apply_pag_node import PAGAttentionNode
## Temporal
# from .nodes.apply_rave_node import ApplyFluxRaveAttentionNode
# from .nodes.add_flow_flux_node import AddFluxFlowNode
## Reference-Only
# from .nodes.apply_ref_flux import ApplyRefFluxNode
## Regional
from .nodes.regional_cond_nodes import ApplyRegionalCondsNode, CreateRegionalCondNode
from .nodes.redux_apply_node import RegionalStyleModelApplyNode
# General
from .nodes.attn_override_node import FluxAttnOverrideNode, RFSingleBlocksOverrideNode, RFDoubleBlocksOverrideNode
NODE_CLASS_MAPPINGS = {
"ConfigureModifiedFlux": ConfigureModifiedFluxNode,
## Inversion
"InFluxFlipSigmas": InFluxFlipSigmasNode,
"InFluxModelSamplingPred": InFluxModelSamplingPredNode,
"OutFluxModelSamplingPred": OutFluxModelSamplingPredNode,
"FluxDeGuidance": FluxDeGuidance,
"FluxInverseSampler": FluxInverseSamplerNode,
"FluxNoiseMixer": FluxNoiseMixerNode,
### RF-Inversion
"FluxForwardODESampler": FluxForwardODESamplerNode,
"FluxReverseODESampler": FluxReverseODESamplerNode,
### RF-Edit
"FlowEditForwardSampler": FlowEditForwardSamplerNode,
"FlowEditReverseSampler": FlowEditReverseSamplerNode,
"PrepareAttnBank": PrepareAttnBankNode,
"RFSingleBlocksOverride": RFSingleBlocksOverrideNode,
"RFDoubleBlocksOverride": RFDoubleBlocksOverrideNode,
## ATTN
"FluxAttnOverride": FluxAttnOverrideNode,
"SEGAttention": SEGAttentionNode,
"PAGAttention": PAGAttentionNode,
## Temporal
# "AddFluxFlow": AddFluxFlowNode,
# "ApplyFluxRaveAttention": ApplyFluxRaveAttentionNode,
## Reference
# "ApplyRefFlux": ApplyRefFluxNode,
## Regional
"ApplyRegionalConds": ApplyRegionalCondsNode,
"CreateRegionalCond": CreateRegionalCondNode,
"RegionalStyleModelApply": RegionalStyleModelApplyNode
}
NODE_DISPLAY_NAME_MAPPINGS = {
"ConfigureModifiedFlux": "Configure Modified Flux",
## Inversion
"InFluxFlipSigmas": "Flip Flux Sigmas",
"InFluxModelSamplingPred": "Inverse Flux Model Pred",
"OutFluxModelSamplingPred": "Outverse Flux Model Pred",
"FluxDeGuidance": "Flux DeGuidance",
"FluxInverseSampler": "Flux Inverse Sampler",
"FluxNoiseMixer": "Flux Mix Noise",
### RF-Inversion
"FluxForwardODESampler": "Flux Forward ODE Sampler",
"FluxReverseODESampler": "Flux Reverse ODE Sampler",
### RF-Edit
"FlowEditForwardSampler": "RF-Edit Forward Sampler",
"FlowEditReverseSampler": "RF-Edit Reverse Sampler",
"PrepareAttnBank": "RF-Edit Prep Attn Inj",
"RFSingleBlocksOverride": "RF-Edit Single Layers Override",
"RFDoubleBlocksOverride": "RF-Edit Double Layers Override",
## Enhance
"FluxAttnOverride": "Flux Attention Override",
"SEGAttention": "Apply Flux SEG Attention",
"PAGAttention": "Apply Flux PAG Attention",
## Temporal
# "AddFluxFlow": "Add Flux Flow",
# "ApplyFluxRaveAttention": "Apply Flux Rave Attn",
## Reference
# "ApplyRefFlux": "Apply Reference-Only",
## Regional
"ApplyRegionalConds": "Apply Flux Regional Conds",
"CreateRegionalCond": "Create Flux Regional Cond",
"RegionalStyleModelApply": "Regional Style Model Apply",
}