File size: 8,488 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
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
import comfy.sample

from .nodes_main import (ControlNetLoaderAdvanced, DiffControlNetLoaderAdvanced,
                         AdvancedControlNetApply, AdvancedControlNetApplySingle)
from .nodes_weight import (DefaultWeights, ScaledSoftMaskedUniversalWeights, ScaledSoftUniversalWeights,
                           SoftControlNetWeightsSD15, CustomControlNetWeightsSD15, CustomControlNetWeightsFlux,
                           SoftT2IAdapterWeights, CustomT2IAdapterWeights, ExtrasMiddleMultNode)
from .nodes_keyframes import (LatentKeyframeGroupNode, LatentKeyframeInterpolationNode, LatentKeyframeBatchedGroupNode, LatentKeyframeNode,
                              TimestepKeyframeNode, TimestepKeyframeInterpolationNode, TimestepKeyframeFromStrengthListNode)
from .nodes_sparsectrl import SparseCtrlMergedLoaderAdvanced, SparseCtrlLoaderAdvanced, SparseIndexMethodNode, SparseSpreadMethodNode, RgbSparseCtrlPreprocessor, SparseWeightExtras
from .nodes_reference import ReferenceControlNetNode, ReferenceControlFinetune, ReferencePreprocessorNode
from .nodes_plusplus import PlusPlusLoaderAdvanced, PlusPlusLoaderSingle, PlusPlusInputNode
from .nodes_ctrlora import CtrLoRALoader
from .nodes_loosecontrol import ControlNetLoaderWithLoraAdvanced
from .nodes_deprecated import (LoadImagesFromDirectory, ScaledSoftUniversalWeightsDeprecated,
                               SoftControlNetWeightsDeprecated, CustomControlNetWeightsDeprecated, 
                               SoftT2IAdapterWeightsDeprecated, CustomT2IAdapterWeightsDeprecated,
                               AdvancedControlNetApplyDEPR, AdvancedControlNetApplySingleDEPR,
                               ControlNetLoaderAdvancedDEPR, DiffControlNetLoaderAdvancedDEPR)
from .logger import logger


# NODE MAPPING
NODE_CLASS_MAPPINGS = {
    # Keyframes
    "TimestepKeyframe": TimestepKeyframeNode,
    "ACN_TimestepKeyframeInterpolation": TimestepKeyframeInterpolationNode,
    "ACN_TimestepKeyframeFromStrengthList": TimestepKeyframeFromStrengthListNode,
    "LatentKeyframe": LatentKeyframeNode,
    "LatentKeyframeTiming": LatentKeyframeInterpolationNode,
    "LatentKeyframeBatchedGroup": LatentKeyframeBatchedGroupNode,
    "LatentKeyframeGroup": LatentKeyframeGroupNode,
    # Conditioning
    "ACN_AdvancedControlNetApply_v2": AdvancedControlNetApply,
    "ACN_AdvancedControlNetApplySingle_v2": AdvancedControlNetApplySingle,
    # Loaders
    "ACN_ControlNetLoaderAdvanced": ControlNetLoaderAdvanced,
    "ACN_DiffControlNetLoaderAdvanced": DiffControlNetLoaderAdvanced,
    # Weights
    "ACN_ScaledSoftControlNetWeights": ScaledSoftUniversalWeights,
    "ScaledSoftMaskedUniversalWeights": ScaledSoftMaskedUniversalWeights,
    "ACN_SoftControlNetWeightsSD15": SoftControlNetWeightsSD15,
    "ACN_CustomControlNetWeightsSD15": CustomControlNetWeightsSD15,
    "ACN_CustomControlNetWeightsFlux": CustomControlNetWeightsFlux,
    "ACN_SoftT2IAdapterWeights": SoftT2IAdapterWeights,
    "ACN_CustomT2IAdapterWeights": CustomT2IAdapterWeights,
    "ACN_DefaultUniversalWeights": DefaultWeights,
    "ACN_ExtrasMiddleMult": ExtrasMiddleMultNode,
    # SparseCtrl
    "ACN_SparseCtrlRGBPreprocessor": RgbSparseCtrlPreprocessor,
    "ACN_SparseCtrlLoaderAdvanced": SparseCtrlLoaderAdvanced,
    "ACN_SparseCtrlMergedLoaderAdvanced": SparseCtrlMergedLoaderAdvanced,
    "ACN_SparseCtrlIndexMethodNode": SparseIndexMethodNode,
    "ACN_SparseCtrlSpreadMethodNode": SparseSpreadMethodNode,
    "ACN_SparseCtrlWeightExtras": SparseWeightExtras,
    # ControlNet++
    "ACN_ControlNet++LoaderSingle": PlusPlusLoaderSingle,
    "ACN_ControlNet++LoaderAdvanced": PlusPlusLoaderAdvanced,
    "ACN_ControlNet++InputNode": PlusPlusInputNode,
    # CtrLoRA
    "ACN_CtrLoRALoader": CtrLoRALoader,
    # Reference
    "ACN_ReferencePreprocessor": ReferencePreprocessorNode,
    "ACN_ReferenceControlNet": ReferenceControlNetNode,
    "ACN_ReferenceControlNetFinetune": ReferenceControlFinetune,
    # LOOSEControl
    #"ACN_ControlNetLoaderWithLoraAdvanced": ControlNetLoaderWithLoraAdvanced,
    # Deprecated
    "LoadImagesFromDirectory": LoadImagesFromDirectory,
    "ScaledSoftControlNetWeights": ScaledSoftUniversalWeightsDeprecated,
    "SoftControlNetWeights": SoftControlNetWeightsDeprecated,
    "CustomControlNetWeights": CustomControlNetWeightsDeprecated,
    "SoftT2IAdapterWeights": SoftT2IAdapterWeightsDeprecated,
    "CustomT2IAdapterWeights": CustomT2IAdapterWeightsDeprecated,
    "ACN_AdvancedControlNetApply": AdvancedControlNetApplyDEPR,
    "ACN_AdvancedControlNetApplySingle": AdvancedControlNetApplySingleDEPR,
    "ControlNetLoaderAdvanced": ControlNetLoaderAdvancedDEPR,
    "DiffControlNetLoaderAdvanced": DiffControlNetLoaderAdvancedDEPR,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    # Keyframes
    "TimestepKeyframe": "Timestep Keyframe πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_TimestepKeyframeInterpolation": "Timestep Keyframe Interp. πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_TimestepKeyframeFromStrengthList": "Timestep Keyframe From List πŸ›‚πŸ…πŸ…’πŸ…",
    "LatentKeyframe": "Latent Keyframe πŸ›‚πŸ…πŸ…’πŸ…",
    "LatentKeyframeTiming": "Latent Keyframe Interp. πŸ›‚πŸ…πŸ…’πŸ…",
    "LatentKeyframeBatchedGroup": "Latent Keyframe From List πŸ›‚πŸ…πŸ…’πŸ…",
    "LatentKeyframeGroup": "Latent Keyframe Group πŸ›‚πŸ…πŸ…’πŸ…",
    # Conditioning
    "ACN_AdvancedControlNetApply_v2": "Apply Advanced ControlNet πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_AdvancedControlNetApplySingle_v2": "Apply Advanced ControlNet(1) πŸ›‚πŸ…πŸ…’πŸ…",
    # Loaders
    "ACN_ControlNetLoaderAdvanced": "Load Advanced ControlNet Model πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_DiffControlNetLoaderAdvanced": "Load Advanced ControlNet Model (diff) πŸ›‚πŸ…πŸ…’πŸ…",
    # Weights
    "ACN_ScaledSoftControlNetWeights": "Scaled Soft Weights πŸ›‚πŸ…πŸ…’πŸ…",
    "ScaledSoftMaskedUniversalWeights": "Scaled Soft Masked Weights πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_SoftControlNetWeightsSD15": "ControlNet Soft Weights [SD1.5] πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_CustomControlNetWeightsSD15": "ControlNet Custom Weights [SD1.5] πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_CustomControlNetWeightsFlux": "ControlNet Custom Weights [Flux] πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_SoftT2IAdapterWeights": "T2IAdapter Soft Weights πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_CustomT2IAdapterWeights": "T2IAdapter Custom Weights πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_DefaultUniversalWeights": "Default Weights πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_ExtrasMiddleMult": "Middle Weight Extras πŸ›‚πŸ…πŸ…’πŸ…",
    # SparseCtrl
    "ACN_SparseCtrlRGBPreprocessor": "RGB SparseCtrl πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_SparseCtrlLoaderAdvanced": "Load SparseCtrl Model πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_SparseCtrlMergedLoaderAdvanced": "πŸ§ͺLoad Merged SparseCtrl Model πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_SparseCtrlIndexMethodNode": "SparseCtrl Index Method πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_SparseCtrlSpreadMethodNode": "SparseCtrl Spread Method πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_SparseCtrlWeightExtras": "SparseCtrl Weight Extras πŸ›‚πŸ…πŸ…’πŸ…",
    # ControlNet++
    "ACN_ControlNet++LoaderSingle": "Load ControlNet++ Model (Single) πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_ControlNet++LoaderAdvanced": "Load ControlNet++ Model (Multi) πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_ControlNet++InputNode": "ControlNet++ Input πŸ›‚πŸ…πŸ…’πŸ…",
    # CtrLoRA
    "ACN_CtrLoRALoader": "Load CtrLoRA Model πŸ›‚πŸ…πŸ…’πŸ…",
    # Reference
    "ACN_ReferencePreprocessor": "Reference Preproccessor πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_ReferenceControlNet": "Reference ControlNet πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_ReferenceControlNetFinetune": "Reference ControlNet (Finetune) πŸ›‚πŸ…πŸ…’πŸ…",
    # LOOSEControl
    #"ACN_ControlNetLoaderWithLoraAdvanced": "Load Adv. ControlNet Model w/ LoRA πŸ›‚πŸ…πŸ…’πŸ…",
    # Deprecated
    "LoadImagesFromDirectory": "🚫Load Images [DEPRECATED] πŸ›‚πŸ…πŸ…’πŸ…",
    "ScaledSoftControlNetWeights": "Scaled Soft Weights πŸ›‚πŸ…πŸ…’πŸ…",
    "SoftControlNetWeights": "ControlNet Soft Weights πŸ›‚πŸ…πŸ…’πŸ…",
    "CustomControlNetWeights": "ControlNet Custom Weights πŸ›‚πŸ…πŸ…’πŸ…",
    "SoftT2IAdapterWeights": "T2IAdapter Soft Weights πŸ›‚πŸ…πŸ…’πŸ…",
    "CustomT2IAdapterWeights": "T2IAdapter Custom Weights πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_AdvancedControlNetApply": "Apply Advanced ControlNet πŸ›‚πŸ…πŸ…’πŸ…",
    "ACN_AdvancedControlNetApplySingle": "Apply Advanced ControlNet(1) πŸ›‚πŸ…πŸ…’πŸ…",
    "ControlNetLoaderAdvanced": "Load Advanced ControlNet Model πŸ›‚πŸ…πŸ…’πŸ…",
    "DiffControlNetLoaderAdvanced": "Load Advanced ControlNet Model (diff) πŸ›‚πŸ…πŸ…’πŸ…",
}