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) ππ
π
π
",
}
|