329 lines
11 KiB
Python
329 lines
11 KiB
Python
from comfy_api.latest import ComfyExtension, io
|
|
|
|
from .nodes_ad_settings import (
|
|
AnimateDiffSettingsNode,
|
|
FullStretchPENode,
|
|
ManualAdjustPENode,
|
|
SweetspotStretchPENode,
|
|
WeightAdjustAllAddNode,
|
|
WeightAdjustAllMultNode,
|
|
WeightAdjustIndivAddNode,
|
|
WeightAdjustIndivAttnAddNode,
|
|
WeightAdjustIndivAttnMultNode,
|
|
WeightAdjustIndivMultNode,
|
|
)
|
|
from .nodes_animatelcmi2v import (
|
|
ApplyAnimateLCMI2VModel,
|
|
LoadAnimateDiffAndInjectI2VNode,
|
|
LoadAnimateLCMI2VModelNode,
|
|
UpscaleAndVaeEncode,
|
|
)
|
|
from .nodes_cameractrl import (
|
|
ApplyAnimateDiffWithCameraCtrl,
|
|
CameraCtrlADKeyframeNode,
|
|
CameraCtrlManualAppendPose,
|
|
CameraCtrlPoseAdvanced,
|
|
CameraCtrlPoseBasic,
|
|
CameraCtrlPoseCombo,
|
|
CameraCtrlReplaceCameraParameters,
|
|
CameraCtrlSetOriginalAspectRatio,
|
|
LoadAnimateDiffModelWithCameraCtrl,
|
|
LoadCameraPosesFromFile,
|
|
LoadCameraPosesFromPath,
|
|
)
|
|
from .nodes_conditioning import (
|
|
CombineLoraHookEightOptionalDEPR,
|
|
CombineLoraHookFourOptionalDEPR,
|
|
CombineLoraHooksDEPR,
|
|
ConditioningCombineDEPR,
|
|
ConditioningSetMaskAndCombineHookedDEPR,
|
|
ConditioningSetMaskHookedDEPR,
|
|
ConditioningSetUnmaskedAndCombineHookedDEPR,
|
|
ConditioningTimestepsNodeDEPR,
|
|
CreateLoraHookKeyframeDEPR,
|
|
CreateLoraHookKeyframeFromStrengthListDEPR,
|
|
CreateLoraHookKeyframeInterpolationDEPR,
|
|
MaskableLoraLoaderDEPR,
|
|
MaskableLoraLoaderModelOnlyDEPR,
|
|
MaskableSDModelLoaderDEPR,
|
|
MaskableSDModelLoaderModelOnlyDEPR,
|
|
PairedConditioningCombineDEPR,
|
|
PairedConditioningSetMaskAndCombineHookedDEPR,
|
|
PairedConditioningSetMaskHookedDEPR,
|
|
PairedConditioningSetUnmaskedAndCombineHookedDEPR,
|
|
SetClipLoraHookDEPR,
|
|
SetLoraHookKeyframesDEPR,
|
|
SetModelLoraHookDEPR,
|
|
)
|
|
from .nodes_context import (
|
|
BatchedContextOptionsNode,
|
|
LegacyLoopedUniformContextOptionsNode,
|
|
LoopedUniformContextOptionsNode,
|
|
LoopedUniformViewOptionsNode,
|
|
StandardStaticContextOptionsNode,
|
|
StandardStaticViewOptionsNode,
|
|
StandardUniformContextOptionsNode,
|
|
StandardUniformViewOptionsNode,
|
|
ViewAsContextOptionsNode,
|
|
VisualizeContextOptionsK,
|
|
VisualizeContextOptionsKAdv,
|
|
VisualizeContextOptionsSCustom,
|
|
)
|
|
from .nodes_context_extras import (
|
|
ContextExtras_ContextRef,
|
|
ContextExtras_NaiveReuse,
|
|
ContextRef_KeyframeFromListNode,
|
|
ContextRef_KeyframeInterpolationNode,
|
|
ContextRef_KeyframeMultivalNode,
|
|
ContextRef_ModeFirst,
|
|
ContextRef_ModeIndexes,
|
|
ContextRef_ModeSliding,
|
|
ContextRef_TuneAttn,
|
|
ContextRef_TuneAttnAdain,
|
|
NaiveReuse_KeyframeFromListNode,
|
|
NaiveReuse_KeyframeInterpolationNode,
|
|
NaiveReuse_KeyframeMultivalNode,
|
|
SetContextExtrasOnContextOptions,
|
|
)
|
|
from .nodes_deprecated import (
|
|
AnimateDiffCombineDEPR,
|
|
AnimateDiffLoaderAdvancedDEPR,
|
|
AnimateDiffLoaderDEPR,
|
|
AnimateDiffModelSettingsAdvancedAttnStrengthsDEPR,
|
|
AnimateDiffModelSettingsAdvancedDEPR,
|
|
AnimateDiffModelSettingsDEPR,
|
|
AnimateDiffModelSettingsSimpleDEPR,
|
|
LegacyAnimateDiffLoaderWithContextDEPR,
|
|
)
|
|
from .nodes_extras import (
|
|
AnimateDiffUnload,
|
|
CheckpointLoaderSimpleWithNoiseSelect,
|
|
EmptyLatentImageLarge,
|
|
PerturbedAttentionGuidanceMultival,
|
|
RescaleCFGMultival,
|
|
)
|
|
from .nodes_gen1 import AnimateDiffLoaderGen1
|
|
from .nodes_gen2 import (
|
|
ADKeyframeNode,
|
|
ApplyAnimateDiffModelBasicNode,
|
|
ApplyAnimateDiffModelNode,
|
|
LoadAnimateDiffModelNode,
|
|
UseEvolvedSamplingNode,
|
|
)
|
|
from .nodes_lora import AnimateDiffLoraLoader
|
|
from .nodes_multival import (
|
|
MultivalConvertToMaskNode,
|
|
MultivalDynamicFloatInputNode,
|
|
MultivalDynamicFloatsNode,
|
|
MultivalDynamicNode,
|
|
MultivalScaledMaskNode,
|
|
)
|
|
from .nodes_per_block import (
|
|
ADBlockComboNode,
|
|
ADBlockIndivNode,
|
|
PerBlockHighLevelNode,
|
|
PerBlock_SD15_FromFloatsNode,
|
|
PerBlock_SD15_LowLevelNode,
|
|
PerBlock_SD15_MidLevelNode,
|
|
PerBlock_SDXL_FromFloatsNode,
|
|
PerBlock_SDXL_LowLevelNode,
|
|
PerBlock_SDXL_MidLevelNode,
|
|
)
|
|
from .nodes_pia import (
|
|
ApplyAnimateDiffPIAModel,
|
|
InputPIA_MultivalNode,
|
|
InputPIA_PaperPresetsNode,
|
|
LoadAnimateDiffAndInjectPIANode,
|
|
PIA_ADKeyframeNode,
|
|
)
|
|
from .nodes_sample import (
|
|
AncestralOptionsNode,
|
|
CFGExtrasPAGNode,
|
|
CFGExtrasPAGSimpleNode,
|
|
CFGExtrasRescaleCFGNode,
|
|
CFGExtrasRescaleCFGSimpleNode,
|
|
CustomCFGKeyframeFromListNode,
|
|
CustomCFGKeyframeInterpolationNode,
|
|
CustomCFGKeyframeNode,
|
|
CustomCFGKeyframeSimpleNode,
|
|
CustomCFGNode,
|
|
CustomCFGSimpleNode,
|
|
FreeInitOptionsNode,
|
|
IterationOptionsNode,
|
|
NoisedImageInjectionNode,
|
|
NoisedImageInjectOptionsNode,
|
|
NoiseLayerAddNode,
|
|
NoiseLayerAddWeightedNode,
|
|
NoiseLayerNormalizedSumNode,
|
|
NoiseLayerReplaceNode,
|
|
SampleSettingsNode,
|
|
)
|
|
from .nodes_scheduling import (
|
|
AddValuesReplaceNode,
|
|
ConditionExtractionNode,
|
|
FloatToFloatsNode,
|
|
PromptSchedulingLatentsNode,
|
|
PromptSchedulingNode,
|
|
ValueSchedulingLatentsNode,
|
|
ValueSchedulingNode,
|
|
)
|
|
from .nodes_sigma_schedule import (
|
|
InterpolatedWeightedAverageSigmaScheduleNode,
|
|
RawSigmaScheduleNode,
|
|
SigmaScheduleNode,
|
|
SigmaScheduleToSigmasNode,
|
|
SplitAndCombineSigmaScheduleNode,
|
|
WeightedAverageSigmaScheduleNode,
|
|
)
|
|
|
|
|
|
class AnimateDiffExtension(ComfyExtension):
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [
|
|
AnimateDiffLoraLoader,
|
|
SampleSettingsNode,
|
|
ADKeyframeNode,
|
|
MultivalDynamicNode,
|
|
MultivalDynamicFloatInputNode,
|
|
MultivalDynamicFloatsNode,
|
|
MultivalScaledMaskNode,
|
|
MultivalConvertToMaskNode,
|
|
StandardStaticContextOptionsNode,
|
|
StandardUniformContextOptionsNode,
|
|
LoopedUniformContextOptionsNode,
|
|
ViewAsContextOptionsNode,
|
|
BatchedContextOptionsNode,
|
|
LegacyLoopedUniformContextOptionsNode,
|
|
VisualizeContextOptionsK,
|
|
VisualizeContextOptionsKAdv,
|
|
VisualizeContextOptionsSCustom,
|
|
StandardStaticViewOptionsNode,
|
|
StandardUniformViewOptionsNode,
|
|
LoopedUniformViewOptionsNode,
|
|
SetContextExtrasOnContextOptions,
|
|
ContextExtras_ContextRef,
|
|
ContextRef_ModeFirst,
|
|
ContextRef_ModeSliding,
|
|
ContextRef_ModeIndexes,
|
|
ContextRef_TuneAttn,
|
|
ContextRef_TuneAttnAdain,
|
|
ContextRef_KeyframeMultivalNode,
|
|
ContextRef_KeyframeInterpolationNode,
|
|
ContextRef_KeyframeFromListNode,
|
|
ContextExtras_NaiveReuse,
|
|
NaiveReuse_KeyframeMultivalNode,
|
|
NaiveReuse_KeyframeInterpolationNode,
|
|
NaiveReuse_KeyframeFromListNode,
|
|
IterationOptionsNode,
|
|
FreeInitOptionsNode,
|
|
MaskableLoraLoaderDEPR,
|
|
MaskableLoraLoaderModelOnlyDEPR,
|
|
MaskableSDModelLoaderDEPR,
|
|
MaskableSDModelLoaderModelOnlyDEPR,
|
|
CombineLoraHooksDEPR,
|
|
CombineLoraHookFourOptionalDEPR,
|
|
CombineLoraHookEightOptionalDEPR,
|
|
SetLoraHookKeyframesDEPR,
|
|
SetClipLoraHookDEPR,
|
|
CreateLoraHookKeyframeDEPR,
|
|
CreateLoraHookKeyframeInterpolationDEPR,
|
|
CreateLoraHookKeyframeFromStrengthListDEPR,
|
|
SetModelLoraHookDEPR,
|
|
PairedConditioningSetMaskHookedDEPR,
|
|
ConditioningSetMaskHookedDEPR,
|
|
PairedConditioningSetMaskAndCombineHookedDEPR,
|
|
ConditioningSetMaskAndCombineHookedDEPR,
|
|
PairedConditioningSetUnmaskedAndCombineHookedDEPR,
|
|
ConditioningSetUnmaskedAndCombineHookedDEPR,
|
|
PairedConditioningCombineDEPR,
|
|
ConditioningCombineDEPR,
|
|
ConditioningTimestepsNodeDEPR,
|
|
NoiseLayerAddNode,
|
|
NoiseLayerAddWeightedNode,
|
|
NoiseLayerNormalizedSumNode,
|
|
NoiseLayerReplaceNode,
|
|
AnimateDiffSettingsNode,
|
|
SweetspotStretchPENode,
|
|
FullStretchPENode,
|
|
ManualAdjustPENode,
|
|
WeightAdjustAllAddNode,
|
|
WeightAdjustAllMultNode,
|
|
WeightAdjustIndivAddNode,
|
|
WeightAdjustIndivMultNode,
|
|
WeightAdjustIndivAttnAddNode,
|
|
WeightAdjustIndivAttnMultNode,
|
|
CustomCFGSimpleNode,
|
|
CustomCFGNode,
|
|
CustomCFGKeyframeSimpleNode,
|
|
CustomCFGKeyframeNode,
|
|
CustomCFGKeyframeInterpolationNode,
|
|
CustomCFGKeyframeFromListNode,
|
|
CFGExtrasPAGSimpleNode,
|
|
CFGExtrasPAGNode,
|
|
CFGExtrasRescaleCFGSimpleNode,
|
|
CFGExtrasRescaleCFGNode,
|
|
SigmaScheduleNode,
|
|
RawSigmaScheduleNode,
|
|
WeightedAverageSigmaScheduleNode,
|
|
InterpolatedWeightedAverageSigmaScheduleNode,
|
|
SplitAndCombineSigmaScheduleNode,
|
|
SigmaScheduleToSigmasNode,
|
|
NoisedImageInjectionNode,
|
|
NoisedImageInjectOptionsNode,
|
|
AncestralOptionsNode,
|
|
PromptSchedulingNode,
|
|
PromptSchedulingLatentsNode,
|
|
ValueSchedulingNode,
|
|
ValueSchedulingLatentsNode,
|
|
ConditionExtractionNode,
|
|
AddValuesReplaceNode,
|
|
FloatToFloatsNode,
|
|
ADBlockComboNode,
|
|
ADBlockIndivNode,
|
|
PerBlockHighLevelNode,
|
|
PerBlock_SD15_MidLevelNode,
|
|
PerBlock_SD15_LowLevelNode,
|
|
PerBlock_SD15_FromFloatsNode,
|
|
PerBlock_SDXL_MidLevelNode,
|
|
PerBlock_SDXL_LowLevelNode,
|
|
PerBlock_SDXL_FromFloatsNode,
|
|
AnimateDiffUnload,
|
|
EmptyLatentImageLarge,
|
|
CheckpointLoaderSimpleWithNoiseSelect,
|
|
PerturbedAttentionGuidanceMultival,
|
|
RescaleCFGMultival,
|
|
AnimateDiffLoaderGen1,
|
|
UseEvolvedSamplingNode,
|
|
ApplyAnimateDiffModelBasicNode,
|
|
ApplyAnimateDiffModelNode,
|
|
LoadAnimateDiffModelNode,
|
|
ApplyAnimateLCMI2VModel,
|
|
LoadAnimateLCMI2VModelNode,
|
|
UpscaleAndVaeEncode,
|
|
LoadAnimateDiffAndInjectI2VNode,
|
|
ApplyAnimateDiffWithCameraCtrl,
|
|
LoadAnimateDiffModelWithCameraCtrl,
|
|
CameraCtrlADKeyframeNode,
|
|
LoadCameraPosesFromFile,
|
|
LoadCameraPosesFromPath,
|
|
CameraCtrlPoseBasic,
|
|
CameraCtrlPoseCombo,
|
|
CameraCtrlPoseAdvanced,
|
|
CameraCtrlManualAppendPose,
|
|
CameraCtrlReplaceCameraParameters,
|
|
CameraCtrlSetOriginalAspectRatio,
|
|
ApplyAnimateDiffPIAModel,
|
|
InputPIA_MultivalNode,
|
|
InputPIA_PaperPresetsNode,
|
|
PIA_ADKeyframeNode,
|
|
LoadAnimateDiffAndInjectPIANode,
|
|
LegacyAnimateDiffLoaderWithContextDEPR,
|
|
AnimateDiffLoaderDEPR,
|
|
AnimateDiffLoaderAdvancedDEPR,
|
|
AnimateDiffCombineDEPR,
|
|
AnimateDiffModelSettingsDEPR,
|
|
AnimateDiffModelSettingsSimpleDEPR,
|
|
AnimateDiffModelSettingsAdvancedDEPR,
|
|
AnimateDiffModelSettingsAdvancedAttnStrengthsDEPR,
|
|
]
|