74 lines
3.7 KiB
Python
74 lines
3.7 KiB
Python
from comfy_api.latest import ComfyExtension, io
|
|
|
|
from .nodes_main import (ControlNetLoaderAdvanced, DiffControlNetLoaderAdvanced, AnimaLLLiteLoaderAdvanced,
|
|
AdvancedControlNetApply, AdvancedControlNetApplySingle)
|
|
from .nodes_weight import (DefaultWeights, ScaledSoftMaskedUniversalWeights, ScaledSoftUniversalWeights,
|
|
SoftControlNetWeightsSD15, CustomControlNetWeightsSD15, CustomControlNetWeightsFlux,
|
|
CustomControlNetWeightsAnima, SoftT2IAdapterWeights, CustomT2IAdapterWeights, ExtrasMiddleMultNode,
|
|
AnimaLLLiteExtras)
|
|
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_deprecated import (LoadImagesFromDirectory, ScaledSoftUniversalWeightsDeprecated,
|
|
SoftControlNetWeightsDeprecated, CustomControlNetWeightsDeprecated,
|
|
SoftT2IAdapterWeightsDeprecated, CustomT2IAdapterWeightsDeprecated,
|
|
AdvancedControlNetApplyDEPR, AdvancedControlNetApplySingleDEPR,
|
|
ControlNetLoaderAdvancedDEPR, DiffControlNetLoaderAdvancedDEPR)
|
|
|
|
|
|
|
|
|
|
class AdvancedControlNetExtension(ComfyExtension):
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
TimestepKeyframeNode,
|
|
TimestepKeyframeInterpolationNode,
|
|
TimestepKeyframeFromStrengthListNode,
|
|
LatentKeyframeNode,
|
|
LatentKeyframeInterpolationNode,
|
|
LatentKeyframeBatchedGroupNode,
|
|
LatentKeyframeGroupNode,
|
|
AdvancedControlNetApply,
|
|
AdvancedControlNetApplySingle,
|
|
ControlNetLoaderAdvanced,
|
|
DiffControlNetLoaderAdvanced,
|
|
AnimaLLLiteLoaderAdvanced,
|
|
ScaledSoftUniversalWeights,
|
|
ScaledSoftMaskedUniversalWeights,
|
|
SoftControlNetWeightsSD15,
|
|
CustomControlNetWeightsSD15,
|
|
CustomControlNetWeightsFlux,
|
|
CustomControlNetWeightsAnima,
|
|
SoftT2IAdapterWeights,
|
|
CustomT2IAdapterWeights,
|
|
DefaultWeights,
|
|
ExtrasMiddleMultNode,
|
|
AnimaLLLiteExtras,
|
|
RgbSparseCtrlPreprocessor,
|
|
SparseCtrlLoaderAdvanced,
|
|
SparseCtrlMergedLoaderAdvanced,
|
|
SparseIndexMethodNode,
|
|
SparseSpreadMethodNode,
|
|
SparseWeightExtras,
|
|
PlusPlusLoaderSingle,
|
|
PlusPlusLoaderAdvanced,
|
|
PlusPlusInputNode,
|
|
CtrLoRALoader,
|
|
ReferencePreprocessorNode,
|
|
ReferenceControlNetNode,
|
|
ReferenceControlFinetune,
|
|
LoadImagesFromDirectory,
|
|
ScaledSoftUniversalWeightsDeprecated,
|
|
SoftControlNetWeightsDeprecated,
|
|
CustomControlNetWeightsDeprecated,
|
|
SoftT2IAdapterWeightsDeprecated,
|
|
CustomT2IAdapterWeightsDeprecated,
|
|
AdvancedControlNetApplyDEPR,
|
|
AdvancedControlNetApplySingleDEPR,
|
|
ControlNetLoaderAdvancedDEPR,
|
|
DiffControlNetLoaderAdvancedDEPR
|
|
]
|