709 lines
28 KiB
Python
709 lines
28 KiB
Python
# ==========================================================
|
||
# __init__.py — Registro nodi IAMCCS
|
||
# ==========================================================
|
||
|
||
import logging
|
||
import os
|
||
|
||
from .iamccs_comfy_compat import apply_iamccs_comfy_compat_patches
|
||
|
||
apply_iamccs_comfy_compat_patches()
|
||
|
||
# ComfyUI frontend assets
|
||
WEB_DIRECTORY = "web"
|
||
|
||
from .iamccs_wan_lora_stack import (
|
||
IAMCCS_WanLoRAStack,
|
||
IAMCCS_ModelWithLoRA,
|
||
)
|
||
from .iamccs_wan_lora_stack_simple import (
|
||
IAMCCS_WanLoRAStackModelIO,
|
||
)
|
||
from .iamccs_wan_lora_schedule import (
|
||
IAMCCS_WanLoRASchedule,
|
||
)
|
||
|
||
from .iamccs_ltx2_lora_stack import (
|
||
IAMCCS_LTX2_LoRAStack,
|
||
IAMCCS_LTX2_LoRAStackStaged,
|
||
IAMCCS_ModelWithLoRA_LTX2,
|
||
IAMCCS_ModelWithLoRA_LTX2_Staged,
|
||
IAMCCS_LTX2_LoRAStackModelIO,
|
||
)
|
||
|
||
from .iamccs_ltx2_lora_stack_segmented6 import (
|
||
IAMCCS_LTX2_LoRAStackSegmented6,
|
||
IAMCCS_LTX2_ModelWithLoRA_Segmented6,
|
||
)
|
||
|
||
from .iamccs_ltx2_tools import (
|
||
IAMCCS_LTX2_FrameRateSync,
|
||
IAMCCS_LTX2_Validator,
|
||
IAMCCS_LTX2_TimeFrameCount,
|
||
IAMCCS_LTX2_EnsureFrames8nPlus1,
|
||
IAMCCS_LTX2_EnsureMinFrames,
|
||
IAMCCS_LTX2_ControlPreprocess,
|
||
IAMCCS_LTX2_ImageBatchPadReflect,
|
||
IAMCCS_LTX2_ImageBatchCropByPad,
|
||
IAMCCS_SegmentPlanner,
|
||
IAMCCS_SegmentPlanFromPlanner,
|
||
IAMCCS_SourceRangeFromSegmentPlan,
|
||
IAMCCS_TwoSegmentPlanner,
|
||
IAMCCS_ThreeSegmentPlanner,
|
||
IAMCCS_SegmentSwitch,
|
||
)
|
||
|
||
from .iamccs_ltx2_extension_module import (
|
||
IAMCCS_LTX2_ExtensionModule,
|
||
IAMCCS_LTX2_ExtensionModule_Disk,
|
||
IAMCCS_LoadImagesFromDirLite,
|
||
IAMCCS_SourceFramesToDisk,
|
||
IAMCCS_StartDirToVideoLatent,
|
||
IAMCCS_VideoCombineFromDir,
|
||
IAMCCS_LTX2_ExtensionModule_simple,
|
||
IAMCCS_LTX2_GetImageFromBatch,
|
||
IAMCCS_LTX2_ReferenceImageSwitch,
|
||
IAMCCS_LTX2_ReferenceStartFramesInjector,
|
||
IAMCCS_LTX2_FrameCountValidator,
|
||
IAMCCS_LTX2_FirstLastFramesController,
|
||
IAMCCS_LTX2_ContextLatent,
|
||
IAMCCS_LTX2_MiddleFrames,
|
||
IAMCCS_LTX2_FirstLastLatentControl,
|
||
IAMCCS_LTX2_FirstLastLatentControl_Pro,
|
||
)
|
||
|
||
from .iamccs_ltx2_temporal_overlap_samplers import (
|
||
IAMCCS_LTX2_ConditionNextLatentWithPrevOverlap,
|
||
IAMCCS_LTX2_InitLatentSampler,
|
||
IAMCCS_LTX2_LoopingSampler,
|
||
IAMCCS_LTX2_OneShotLowRAMLooper,
|
||
IAMCCS_LTX2_ExtendSampler,
|
||
)
|
||
|
||
from .iamccs_wan_svipro_motion import (
|
||
IAMCCS_WanImageMotion,
|
||
WanImageMotionPro as WanImageMotionProPlus,
|
||
IAMCCS_WanImageMotionPro_Simple as IAMCCS_WanImageMotionProPlus_Simple,
|
||
IAMCCS_WanImageMotionInductive,
|
||
IAMCCS_WanSVIToFLFBridgePro as IAMCCS_WanSVIToFLFBridgeProPlus,
|
||
IAMCCS_WanSVIToFLFBridgePro_Simple as IAMCCS_WanSVIToFLFBridgeProPlus_Simple,
|
||
WanMotionProTrimmer,
|
||
IAMCCS_WanPrevTailPrep,
|
||
)
|
||
|
||
from .iamccs_wan_long_length import (
|
||
IAMCCS_WanLongPlanner,
|
||
IAMCCS_WanContinuityGuide,
|
||
IAMCCS_WanPromptPhasePlanner,
|
||
IAMCCS_WanPromptLoopInfo,
|
||
IAMCCS_WanIndexedPromptEncode,
|
||
IAMCCS_WanImageBatchFrameSelect,
|
||
)
|
||
|
||
from .iamccs_wan_svipro_motion_legacy import (
|
||
WanImageMotionProLegacy,
|
||
)
|
||
|
||
from .iamccs_autolink import (
|
||
IAMCCS_SetAutoLink,
|
||
IAMCCS_GetAutoLink,
|
||
IAMCCS_AutoLinkConverter,
|
||
IAMCCS_AutoLinkArguments,
|
||
)
|
||
|
||
from .iamccs_gguf_accelerator import (
|
||
IAMCCS_GGUF_accelerator,
|
||
)
|
||
|
||
from .iamccs_sampler_advanced_v1 import (
|
||
IAMCCS_SamplerAdvancedVersion1,
|
||
)
|
||
|
||
from .iamccs_bus_group import (
|
||
IAMCCS_bus_group,
|
||
)
|
||
|
||
from .iamccs_image_resize import (
|
||
IAMCCS_ImageResizeBatchSafe,
|
||
IAMCCS_LoadResizeSegmentFromDir,
|
||
)
|
||
|
||
from .iamccs_multiswitch import (
|
||
IAMCCS_MultiSwitch,
|
||
)
|
||
|
||
from .iamccs_hw_supporter import (
|
||
IAMCCS_HwSupporter,
|
||
IAMCCS_HwSupporterAny,
|
||
IAMCCS_HardMemoryPurge,
|
||
IAMCCS_VRAMCleanup,
|
||
IAMCCS_VRAMFlushLatent,
|
||
IAMCCS_VAEDecodeTiledSafe,
|
||
IAMCCS_VAEDecodeToDisk,
|
||
)
|
||
|
||
from .iamccs_hw_probe_node import (
|
||
IAMCCS_HWProbeRecommendations,
|
||
)
|
||
|
||
from .iamccs_qwen_vl_flf import (
|
||
IAMCCS_QWEN_VL_FLF,
|
||
IAMCCS_QWEN_VL_FLF_Advanced,
|
||
)
|
||
|
||
from .iamccs_move_ahead import (
|
||
IAMCCS_MoveAhead,
|
||
IAMCCS_MoveAheadEnforcer,
|
||
IAMCCS_MotionScale,
|
||
IAMCCS_MotionScaleAdvanced,
|
||
)
|
||
|
||
from .iamccs_motion_bridge import (
|
||
IAMCCS_MotionBridgeSave,
|
||
IAMCCS_MotionBridgeLoad,
|
||
IAMCCS_LatentTailSlice,
|
||
)
|
||
|
||
from .iamccs_audio_extender import (
|
||
IAMCCS_AudioExtensionMath,
|
||
IAMCCS_AudioExtender,
|
||
IAMCCS_AudioTimelineAssembler,
|
||
IAMCCS_AudioTimelineGate,
|
||
)
|
||
|
||
from .iamccs_ltx2_segment_queue import (
|
||
IAMCCS_LTX2_BlendLatentBridge,
|
||
IAMCCS_LTX2_LastFrameBridgeLoad,
|
||
IAMCCS_LTX2_LastFrameBridgeSave,
|
||
IAMCCS_LTX2_LoadLatentBridge,
|
||
IAMCCS_LTX2_LongVideoWrapperPrep,
|
||
IAMCCS_LTX2_LongVideoWrapperPrepDisk,
|
||
IAMCCS_LTX2_SaveLatentBridge,
|
||
IAMCCS_LTX2_SegmentQueueLoop,
|
||
)
|
||
|
||
from .iamccs_image_resize import (
|
||
IAMCCS_ImageResizeBatchSafe,
|
||
)
|
||
|
||
from .iamccs_value_monitor import (
|
||
IAMCCS_IntValueMonitor,
|
||
)
|
||
|
||
from .iamccs_flux_klein_multigen import (
|
||
IAMCCS_FluxKleinMultiGen,
|
||
)
|
||
|
||
from .iamccs_qwen_multigen import (
|
||
IAMCCS_QwenMultiGen,
|
||
)
|
||
|
||
from .iamccs_multiline_prompt_splitter import (
|
||
IAMCCS_MultilinePromptSplitter8,
|
||
)
|
||
|
||
# Nodi principali
|
||
NODE_CLASS_MAPPINGS = {
|
||
"IAMCCS_WanLoRAStack": IAMCCS_WanLoRAStack,
|
||
"IAMCCS_ModelWithLoRA": IAMCCS_ModelWithLoRA,
|
||
"IAMCCS_WanLoRAStackModelIO": IAMCCS_WanLoRAStackModelIO,
|
||
"IAMCCS_WanLoRASchedule": IAMCCS_WanLoRASchedule,
|
||
# Backward-compatible key (kept as-is for existing workflows)
|
||
"iamccs_ltx2_lora_stack": IAMCCS_LTX2_LoRAStack,
|
||
# Preferred explicit names
|
||
"IAMCCS_LTX2_LoRAStack": IAMCCS_LTX2_LoRAStack,
|
||
"IAMCCS_LTX2_LoRAStackStaged": IAMCCS_LTX2_LoRAStackStaged,
|
||
"IAMCCS_ModelWithLoRA_LTX2": IAMCCS_ModelWithLoRA_LTX2,
|
||
"IAMCCS_ModelWithLoRA_LTX2_Staged": IAMCCS_ModelWithLoRA_LTX2_Staged,
|
||
"IAMCCS_LTX2_LoRAStackModelIO": IAMCCS_LTX2_LoRAStackModelIO,
|
||
"IAMCCS_LTX2_LoRAStackSegmented6": IAMCCS_LTX2_LoRAStackSegmented6,
|
||
"IAMCCS_LTX2_ModelWithLoRA_Segmented6": IAMCCS_LTX2_ModelWithLoRA_Segmented6,
|
||
|
||
"IAMCCS_LTX2_FrameRateSync": IAMCCS_LTX2_FrameRateSync,
|
||
"IAMCCS_LTX2_Validator": IAMCCS_LTX2_Validator,
|
||
"IAMCCS_LTX2_TimeFrameCount": IAMCCS_LTX2_TimeFrameCount,
|
||
"IAMCCS_LTX2_EnsureFrames8nPlus1": IAMCCS_LTX2_EnsureFrames8nPlus1,
|
||
"IAMCCS_LTX2_EnsureMinFrames": IAMCCS_LTX2_EnsureMinFrames,
|
||
"IAMCCS_LTX2_ControlPreprocess": IAMCCS_LTX2_ControlPreprocess,
|
||
"IAMCCS_LTX2_ImageBatchPadReflect": IAMCCS_LTX2_ImageBatchPadReflect,
|
||
"IAMCCS_LTX2_ImageBatchCropByPad": IAMCCS_LTX2_ImageBatchCropByPad,
|
||
"IAMCCS_SegmentPlanner": IAMCCS_SegmentPlanner,
|
||
"IAMCCS_SegmentPlanFromPlanner": IAMCCS_SegmentPlanFromPlanner,
|
||
"IAMCCS_SourceRangeFromSegmentPlan": IAMCCS_SourceRangeFromSegmentPlan,
|
||
"IAMCCS_TwoSegmentPlanner": IAMCCS_TwoSegmentPlanner,
|
||
"IAMCCS_ThreeSegmentPlanner": IAMCCS_ThreeSegmentPlanner,
|
||
"IAMCCS_SegmentSwitch": IAMCCS_SegmentSwitch,
|
||
"IAMCCS_LTX2_ExtensionModule": IAMCCS_LTX2_ExtensionModule,
|
||
"IAMCCS_LTX2_ExtensionModule_Disk": IAMCCS_LTX2_ExtensionModule_Disk,
|
||
"IAMCCS_LoadImagesFromDirLite": IAMCCS_LoadImagesFromDirLite,
|
||
"IAMCCS_SourceFramesToDisk": IAMCCS_SourceFramesToDisk,
|
||
"IAMCCS_StartDirToVideoLatent": IAMCCS_StartDirToVideoLatent,
|
||
"IAMCCS_VideoCombineFromDir": IAMCCS_VideoCombineFromDir,
|
||
"IAMCCS_LTX2_ExtensionModule_simple": IAMCCS_LTX2_ExtensionModule_simple,
|
||
"IAMCCS_LTX2_GetImageFromBatch": IAMCCS_LTX2_GetImageFromBatch,
|
||
"IAMCCS_LTX2_ReferenceImageSwitch": IAMCCS_LTX2_ReferenceImageSwitch,
|
||
"IAMCCS_LTX2_ReferenceStartFramesInjector": IAMCCS_LTX2_ReferenceStartFramesInjector,
|
||
"IAMCCS_LTX2_FrameCountValidator": IAMCCS_LTX2_FrameCountValidator,
|
||
"IAMCCS_LTX2_FirstLastFramesController": IAMCCS_LTX2_FirstLastFramesController,
|
||
"IAMCCS_LTX2_ContextLatent": IAMCCS_LTX2_ContextLatent,
|
||
"IAMCCS_LTX2_MiddleFrames": IAMCCS_LTX2_MiddleFrames,
|
||
"IAMCCS_LTX2_FirstLastLatentControl": IAMCCS_LTX2_FirstLastLatentControl,
|
||
"IAMCCS_LTX2_FirstLastLatentControl_Pro": IAMCCS_LTX2_FirstLastLatentControl_Pro,
|
||
"IAMCCS_LTX2_InitLatentSampler": IAMCCS_LTX2_InitLatentSampler,
|
||
"IAMCCS_LTX2_LoopingSampler": IAMCCS_LTX2_LoopingSampler,
|
||
"IAMCCS_LTX2_OneShotLowRAMLooper": IAMCCS_LTX2_OneShotLowRAMLooper,
|
||
"IAMCCS_LTX2_ExtendSampler": IAMCCS_LTX2_ExtendSampler,
|
||
"IAMCCS_LTX2_ConditionNextLatentWithPrevOverlap": IAMCCS_LTX2_ConditionNextLatentWithPrevOverlap,
|
||
"IAMCCS_WanImageMotion": IAMCCS_WanImageMotion,
|
||
# Backward-compat alias: workflow JSONs saved with the _AdaIN name still load.
|
||
"IAMCCS_WanImageMotion_AdaIN": IAMCCS_WanImageMotion,
|
||
"WanImageMotionPro": WanImageMotionProPlus,
|
||
"IAMCCS_WanImageMotionPro_AdaIN": WanImageMotionProPlus,
|
||
# Keep the historic key on the current implementation so existing workflows
|
||
# load the new Plus node with continuity profiles and presets.
|
||
"IAMCCS_WanImageMotionPro": WanImageMotionProPlus,
|
||
# Explicit legacy entrypoint for older raw-only behavior.
|
||
"WanImageMotionProLegacy": WanImageMotionProLegacy,
|
||
"IAMCCS_WanImageMotionProLegacy": WanImageMotionProLegacy,
|
||
"WanImageMotionProPlus": WanImageMotionProPlus,
|
||
"IAMCCS_WanImageMotionProPlus": WanImageMotionProPlus,
|
||
"IAMCCS_WanImageMotionProPlus_Simple": IAMCCS_WanImageMotionProPlus_Simple,
|
||
"IAMCCS_WanImageMotionInductive": IAMCCS_WanImageMotionInductive,
|
||
"IAMCCS_WanSVIToFLFBridgeProPlus": IAMCCS_WanSVIToFLFBridgeProPlus,
|
||
"IAMCCS_WanSVIToFLFBridgeProPlus_Simple": IAMCCS_WanSVIToFLFBridgeProPlus_Simple,
|
||
"WanMotionProTrimmer": WanMotionProTrimmer,
|
||
"IAMCCS_WanPrevTailPrep": IAMCCS_WanPrevTailPrep,
|
||
"IAMCCS_WanLongPlanner": IAMCCS_WanLongPlanner,
|
||
"IAMCCS_WanContinuityGuide": IAMCCS_WanContinuityGuide,
|
||
"IAMCCS_WanPromptPhasePlanner": IAMCCS_WanPromptPhasePlanner,
|
||
"IAMCCS_WanPromptLoopInfo": IAMCCS_WanPromptLoopInfo,
|
||
"IAMCCS_WanIndexedPromptEncode": IAMCCS_WanIndexedPromptEncode,
|
||
"IAMCCS_WanImageBatchFrameSelect": IAMCCS_WanImageBatchFrameSelect,
|
||
|
||
"IAMCCS_SetAutoLink": IAMCCS_SetAutoLink,
|
||
"IAMCCS_GetAutoLink": IAMCCS_GetAutoLink,
|
||
"IAMCCS_AutoLinkConverter": IAMCCS_AutoLinkConverter,
|
||
"IAMCCS_AutoLinkArguments": IAMCCS_AutoLinkArguments,
|
||
|
||
"IAMCCS_GGUF_accelerator": IAMCCS_GGUF_accelerator,
|
||
|
||
"IAMCCS_SamplerAdvancedVersion1": IAMCCS_SamplerAdvancedVersion1,
|
||
|
||
"IAMCCS_bus_group": IAMCCS_bus_group,
|
||
|
||
"IAMCCS_MultiSwitch": IAMCCS_MultiSwitch,
|
||
|
||
"IAMCCS_HwSupporter": IAMCCS_HwSupporter,
|
||
"IAMCCS_HwSupporterAny": IAMCCS_HwSupporterAny,
|
||
"IAMCCS_HardMemoryPurge": IAMCCS_HardMemoryPurge,
|
||
"IAMCCS_VRAMCleanup": IAMCCS_VRAMCleanup,
|
||
"IAMCCS_VRAMFlushLatent": IAMCCS_VRAMFlushLatent,
|
||
"IAMCCS_VAEDecodeTiledSafe": IAMCCS_VAEDecodeTiledSafe,
|
||
"IAMCCS_VAEDecodeToDisk": IAMCCS_VAEDecodeToDisk,
|
||
"IAMCCS_HWProbeRecommendations": IAMCCS_HWProbeRecommendations,
|
||
|
||
"IAMCCS_MoveAhead": IAMCCS_MoveAhead,
|
||
"IAMCCS_MoveAheadEnforcer": IAMCCS_MoveAheadEnforcer,
|
||
"IAMCCS_MotionScale": IAMCCS_MotionScale,
|
||
"IAMCCS_MotionScaleAdvanced": IAMCCS_MotionScaleAdvanced,
|
||
|
||
"IAMCCS_MotionBridgeSave": IAMCCS_MotionBridgeSave,
|
||
"IAMCCS_MotionBridgeLoad": IAMCCS_MotionBridgeLoad,
|
||
"IAMCCS_LatentTailSlice": IAMCCS_LatentTailSlice,
|
||
"IAMCCS_AudioExtensionMath": IAMCCS_AudioExtensionMath,
|
||
"IAMCCS_AudioExtender": IAMCCS_AudioExtender,
|
||
"IAMCCS_AudioTimelineAssembler": IAMCCS_AudioTimelineAssembler,
|
||
"IAMCCS_AudioTimelineGate": IAMCCS_AudioTimelineGate,
|
||
"IAMCCS_LTX2_LastFrameBridgeSave": IAMCCS_LTX2_LastFrameBridgeSave,
|
||
"IAMCCS_LTX2_BlendLatentBridge": IAMCCS_LTX2_BlendLatentBridge,
|
||
"IAMCCS_LTX2_LastFrameBridgeLoad": IAMCCS_LTX2_LastFrameBridgeLoad,
|
||
"IAMCCS_LTX2_LoadLatentBridge": IAMCCS_LTX2_LoadLatentBridge,
|
||
"IAMCCS_LTX2_LongVideoWrapperPrep": IAMCCS_LTX2_LongVideoWrapperPrep,
|
||
"IAMCCS_LTX2_LongVideoWrapperPrepDisk": IAMCCS_LTX2_LongVideoWrapperPrepDisk,
|
||
"IAMCCS_LTX2_SaveLatentBridge": IAMCCS_LTX2_SaveLatentBridge,
|
||
"IAMCCS_LTX2_SegmentQueueLoop": IAMCCS_LTX2_SegmentQueueLoop,
|
||
"IAMCCS_ImageResizeBatchSafe": IAMCCS_ImageResizeBatchSafe,
|
||
"IAMCCS_LoadResizeSegmentFromDir": IAMCCS_LoadResizeSegmentFromDir,
|
||
"IAMCCS_IntValueMonitor": IAMCCS_IntValueMonitor,
|
||
"IAMCCS_QwenMultiGen": IAMCCS_QwenMultiGen,
|
||
"IAMCCS_FluxKleinMultiGen": IAMCCS_FluxKleinMultiGen,
|
||
"IAMCCS_MultilinePromptSplitter8": IAMCCS_MultilinePromptSplitter8,
|
||
|
||
# QwenVL First/Last Frame (registered only if QwenVL is installed)
|
||
**({"IAMCCS_QWEN_VL_FLF": IAMCCS_QWEN_VL_FLF,
|
||
"IAMCCS_QWEN_VL_FLF_Advanced": IAMCCS_QWEN_VL_FLF_Advanced,
|
||
} if IAMCCS_QWEN_VL_FLF is not None else {}),
|
||
|
||
}
|
||
|
||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||
"IAMCCS_WanLoRAStack": "LoRA Stack (WAN-style remap)",
|
||
"IAMCCS_ModelWithLoRA": "Apply LoRA to MODEL (Native)",
|
||
"IAMCCS_WanLoRAStackModelIO": "LoRA Stack (Model In→Out) WAN",
|
||
"IAMCCS_WanLoRASchedule": "LoRA Schedule (WAN, ranged)",
|
||
"iamccs_ltx2_lora_stack": "iamccs_ltx2_lora_stack (3 slots)",
|
||
"IAMCCS_LTX2_LoRAStack": "LoRA Stack (LTX-2, 3 slots)",
|
||
"IAMCCS_LTX2_LoRAStackStaged": "LoRA Stack (LTX-2, staged: stage1+stage2) (BETA)",
|
||
"IAMCCS_ModelWithLoRA_LTX2": "Apply LoRA to MODEL (LTX-2, quiet logs)",
|
||
"IAMCCS_ModelWithLoRA_LTX2_Staged": "Apply LoRA to MODEL (LTX-2, staged) (BETA)",
|
||
"IAMCCS_LTX2_LoRAStackModelIO": "LoRA Stack (Model In→Out) LTX-2",
|
||
"IAMCCS_LTX2_LoRAStackSegmented6": "LoRA Stack (LTX-2, segmented: 3 seg × 2 stages)",
|
||
"IAMCCS_LTX2_ModelWithLoRA_Segmented6": "Apply LoRA to MODEL (LTX-2, segmented: 3 seg × 2 stages)",
|
||
|
||
"IAMCCS_LTX2_FrameRateSync": "LTX-2 FrameRate Sync (int+float)",
|
||
"IAMCCS_LTX2_Validator": "LTX-2 Validator",
|
||
"IAMCCS_LTX2_TimeFrameCount": "LTX-2 TimeFrameCount",
|
||
"IAMCCS_LTX2_EnsureFrames8nPlus1": "LTX-2 Ensure Frames (8n + 1)",
|
||
"IAMCCS_LTX2_EnsureMinFrames": "LTX-2 Ensure Minimum Frames",
|
||
"IAMCCS_LTX2_ControlPreprocess": "LTX-2 Control Preprocess (aux)",
|
||
"IAMCCS_LTX2_ImageBatchPadReflect": "LTX-2 Pad Reflect (IMAGE batch)",
|
||
"IAMCCS_LTX2_ImageBatchCropByPad": "LTX-2 Crop By Pad (IMAGE batch)",
|
||
"IAMCCS_SegmentPlanner": "Segment Planner (song -> LTX frames)",
|
||
"IAMCCS_SegmentPlanFromPlanner": "Segment Plan From Planner (per index)",
|
||
"IAMCCS_SourceRangeFromSegmentPlan": "Source Range From Segment Plan",
|
||
"IAMCCS_TwoSegmentPlanner": "Two Segment Planner (stable 2SEG)",
|
||
"IAMCCS_ThreeSegmentPlanner": "Three Segment Planner (stable 3SEG)",
|
||
"IAMCCS_SegmentSwitch": "Segment Switch (by segment_index)",
|
||
"IAMCCS_LTX2_ExtensionModule": "LTX-2 Extension Module 🎬",
|
||
"IAMCCS_LTX2_ExtensionModule_Disk": "LTX-2 Extension Module (Disk / Low RAM) 💾",
|
||
"IAMCCS_LoadImagesFromDirLite": "Load Images From Dir (Lite) 📁",
|
||
"IAMCCS_SourceFramesToDisk": "Source Frames To Disk 📼💾",
|
||
"IAMCCS_StartDirToVideoLatent": "Start Dir To Video Latent 🚀",
|
||
"IAMCCS_VideoCombineFromDir": "Video Combine From Dir 🎞️",
|
||
"IAMCCS_LTX2_ExtensionModule_simple": "LTX-2 Extension Module (simple) 🎬",
|
||
"IAMCCS_LTX2_GetImageFromBatch": "LTX-2 Get Images From Batch 🎞️",
|
||
"IAMCCS_LTX2_ReferenceImageSwitch": "LTX-2 Reference Image Switch 🧷",
|
||
"IAMCCS_LTX2_ReferenceStartFramesInjector": "LTX-2 Inject Reference Into Start Frames 🧬",
|
||
"IAMCCS_LTX2_FrameCountValidator": "LTX-2 Frame Count Validator ✅ (8n+1)",
|
||
"IAMCCS_LTX2_FirstLastFramesController": "LTX-2 First/Last Frames Controller 🧲",
|
||
"IAMCCS_LTX2_ContextLatent": "LTX-2 Context → Latent (continue) 🧩",
|
||
"IAMCCS_LTX2_MiddleFrames": "LTX-2 Middle Frames (accumulator) 🧷",
|
||
"IAMCCS_LTX2_FirstLastLatentControl": "LTX-2 First/Last → Latent (noise_mask) 🎯",
|
||
"IAMCCS_LTX2_FirstLastLatentControl_Pro": "LTX-2 First/Last → Latent (Pro, slot caps) 🎯",
|
||
"IAMCCS_LTX2_InitLatentSampler": "LTX-2 Init Latent Sampler 🧱",
|
||
"IAMCCS_LTX2_LoopingSampler": "LTX-2 Looping Sampler (temporal overlap) 🧷",
|
||
"IAMCCS_LTX2_OneShotLowRAMLooper": "LTX-2 One-Shot Low-RAM Looper 🪶",
|
||
"IAMCCS_LTX2_ExtendSampler": "LTX-2 Extend Sampler (temporal overlap) 🧷",
|
||
"IAMCCS_LTX2_ConditionNextLatentWithPrevOverlap": "LTX-2 Condition Next Latent (prev overlap) 🧷",
|
||
"IAMCCS_WanImageMotion": "WanImageMotion",
|
||
"IAMCCS_WanImageMotion_AdaIN": "WanImageMotion",
|
||
"WanImageMotionPro": "WanImageMotionPro Plus",
|
||
"IAMCCS_WanImageMotionPro_AdaIN": "WanImageMotionPro Plus",
|
||
"WanImageMotionProLegacy": "WanImageMotionPro Legacy",
|
||
"IAMCCS_WanImageMotionProLegacy": "WanImageMotionPro Legacy",
|
||
"WanImageMotionProPlus": "WanImageMotionPro Plus",
|
||
"IAMCCS_WanImageMotionProPlus": "WanImageMotionPro Plus",
|
||
"IAMCCS_WanImageMotionProPlus_Simple": "WanImageMotionPro Plus Simple",
|
||
"IAMCCS_WanImageMotionInductive": "WanImageMotion Inductive",
|
||
"IAMCCS_WanSVIToFLFBridgeProPlus": "Wan SVI→FLF Bridge Pro Plus",
|
||
"IAMCCS_WanSVIToFLFBridgeProPlus_Simple": "Wan SVI→FLF Bridge Pro Plus",
|
||
"WanMotionProTrimmer": "WanMotionProTrimmer (trim overshoot tail)",
|
||
"IAMCCS_WanPrevTailPrep": "Wan Prev Tail Prep",
|
||
"IAMCCS_WanLongPlanner": "Wan Long Planner",
|
||
"IAMCCS_WanContinuityGuide": "Wan Continuity Guide",
|
||
"IAMCCS_WanPromptPhasePlanner": "Wan Prompt Phase Planner",
|
||
"IAMCCS_WanPromptLoopInfo": "Wan Prompt Loop Info",
|
||
"IAMCCS_WanIndexedPromptEncode": "Wan Indexed Prompt Encode",
|
||
"IAMCCS_WanImageBatchFrameSelect": "Wan Image Batch Frame Select",
|
||
|
||
"IAMCCS_SetAutoLink": "Set AutoLink",
|
||
"IAMCCS_GetAutoLink": "Get AutoLink",
|
||
"IAMCCS_AutoLinkConverter": "AutoLink Converter",
|
||
"IAMCCS_AutoLinkArguments": "AutoLink Arguments",
|
||
|
||
"IAMCCS_GGUF_accelerator": "GGUF Accelerator (patch_on_device)",
|
||
|
||
"IAMCCS_IntValueMonitor": "INT Value Monitor",
|
||
"IAMCCS_QwenMultiGen": "IAMCCS Qwen Multi-Gen",
|
||
"IAMCCS_FluxKleinMultiGen": "Flux Klein Multi-Gen",
|
||
"IAMCCS_MultilinePromptSplitter8": "Multiline Prompt Splitter (8 outputs)",
|
||
|
||
"IAMCCS_SamplerAdvancedVersion1": "Sampler Advanced v1",
|
||
|
||
"IAMCCS_bus_group": "Bus Group (Mute + Solo) (frontend-only)",
|
||
|
||
"IAMCCS_MultiSwitch": "MultiSwitch (dynamic inputs)",
|
||
"IAMCCS_WanSviArgs": "Wan SVI Args",
|
||
"IAMCCS_WanSviChainRunner": "Wan SVI Chain Runner",
|
||
"IAMCCS_WanSviSegmentPick": "Wan SVI Segment Pick",
|
||
|
||
"IAMCCS_HwSupporter": "HW Supporter (auto VRAM/attention/torch knobs)",
|
||
"IAMCCS_HwSupporterAny": "HW Supporter (ANY passthrough)",
|
||
"IAMCCS_HardMemoryPurge": "Hard RAM/VRAM Purge (trim working set)",
|
||
"IAMCCS_VRAMCleanup": "VRAM Cleanup (unload + empty cache)",
|
||
"IAMCCS_VRAMFlushLatent": "VRAM Flush → Latent passthrough (empty cache)",
|
||
"IAMCCS_VAEDecodeTiledSafe": "VAE Decode Tiled (safe, optional cleanup)",
|
||
"IAMCCS_VAEDecodeToDisk": "VAE Decode → Disk (frames, low RAM)",
|
||
"IAMCCS_HWProbeRecommendations": "HW Probe Recommendations (JSON)",
|
||
|
||
"IAMCCS_MoveAhead": "MoveAhead (FreeLong spectral blend) 🎬",
|
||
"IAMCCS_MoveAheadEnforcer": "MoveAhead Enforcer (3-tier motion lock) 🎬",
|
||
"IAMCCS_MotionScale": "MotionScale (temporal RoPE scale) ⚡",
|
||
"IAMCCS_MotionScaleAdvanced": "MotionScale Advanced (RoPE + theta) ⚡",
|
||
|
||
"IAMCCS_MotionBridgeSave": "Motion Bridge Save 🎬💾",
|
||
"IAMCCS_MotionBridgeLoad": "Motion Bridge Load 🎬📂",
|
||
"IAMCCS_LatentTailSlice": "Latent Tail Slice ✂️",
|
||
"IAMCCS_AudioExtensionMath": "Audio Extension Math (timeline sync)",
|
||
"IAMCCS_AudioExtender": "Audio Extender (segment + overlap)",
|
||
"IAMCCS_AudioTimelineAssembler": "Audio Timeline Assembler (full track)",
|
||
"IAMCCS_AudioTimelineGate": "Audio Timeline Gate (continue/stop)",
|
||
"IAMCCS_LTX2_LastFrameBridgeSave": "LTX-2 Last Frame Bridge Save 🖼️💾",
|
||
"IAMCCS_LTX2_BlendLatentBridge": "LTX-2 Blend Latent Bridge 🎚️",
|
||
"IAMCCS_LTX2_LastFrameBridgeLoad": "LTX-2 Last Frame Bridge Load 🖼️",
|
||
"IAMCCS_LTX2_LoadLatentBridge": "LTX-2 Load Latent Bridge 🧬",
|
||
"IAMCCS_LTX2_LongVideoWrapperPrep": "LTX-2 Long Video Wrapper Prep 🧰",
|
||
"IAMCCS_LTX2_LongVideoWrapperPrepDisk": "LTX-2 Long Video Wrapper Prep (Disk) 💾🧰",
|
||
"IAMCCS_LTX2_SaveLatentBridge": "LTX-2 Save Latent Bridge 💾",
|
||
"IAMCCS_LTX2_SegmentQueueLoop": "LTX-2 Segment Queue Loop 🔁",
|
||
"IAMCCS_ImageResizeBatchSafe": "Image Resize Batch Safe (IAMCCS)",
|
||
"IAMCCS_LoadResizeSegmentFromDir": "Load + Resize Segment From Dir 🧰",
|
||
|
||
# QwenVL FLF
|
||
**({"IAMCCS_QWEN_VL_FLF": "QwenVL FLF — First/Last Frame Prompt 🎬",
|
||
"IAMCCS_QWEN_VL_FLF_Advanced": "QwenVL FLF — First/Last Frame Prompt (Advanced) 🎬",
|
||
} if IAMCCS_QWEN_VL_FLF is not None else {}),
|
||
|
||
}
|
||
|
||
WEB_DIRECTORY = "./web"
|
||
|
||
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|
||
|
||
|
||
def _print_startup_banner() -> None:
|
||
# Print once per process.
|
||
if getattr(_print_startup_banner, "_done", False):
|
||
return
|
||
_print_startup_banner._done = True # type: ignore[attr-defined]
|
||
|
||
banner = r"""
|
||
___ _ __ __ ____ ____ ____ ____ _
|
||
|_ _| / \ | \/ |/ ___/ ___|/ ___| | _ \ ___ ___| | _____
|
||
| | / _ \ | |\/| | | | | \___ \ | |_) / _ \ / __| |/ / __|
|
||
| | / ___ \| | | | |__| |___ ___) || __/ (_) | (__| <\__ \
|
||
|___/_/ \_\_| |_|\____\____||____/ |_| \___/ \___|_|\_\___/
|
||
|
||
"""
|
||
log = logging.getLogger("IAMCCS")
|
||
log.info("%s", banner)
|
||
log.info("by IAMCCS (follow me on patreon.com/IAMCCS or carminecristalloscalzi.com)")
|
||
|
||
try:
|
||
keys = sorted(list(NODE_CLASS_MAPPINGS.keys()))
|
||
log.info("IAMCCS nodes loaded: %d", len(keys))
|
||
# Keep log readable: print in chunks.
|
||
chunk = []
|
||
for k in keys:
|
||
chunk.append(k)
|
||
if len(chunk) >= 10:
|
||
log.info("- %s", ", ".join(chunk))
|
||
chunk = []
|
||
if chunk:
|
||
log.info("- %s", ", ".join(chunk))
|
||
except Exception:
|
||
pass
|
||
|
||
|
||
def setup_api_routes() -> None:
|
||
"""IAMCCS API routes used by frontend widgets."""
|
||
|
||
try:
|
||
from server import PromptServer
|
||
from aiohttp import web
|
||
|
||
from .iamccs_hw_probe import recommend_settings
|
||
|
||
routes = PromptServer.instance.routes
|
||
|
||
@routes.get("/api/iamccs/hw_probe")
|
||
async def iamccs_hw_probe_endpoint(request):
|
||
try:
|
||
q = request.rel_url.query
|
||
def _to_int(x):
|
||
try:
|
||
return int(float(x))
|
||
except Exception:
|
||
return None
|
||
def _to_float(x):
|
||
try:
|
||
return float(x)
|
||
except Exception:
|
||
return None
|
||
|
||
width = _to_int(q.get("width"))
|
||
height = _to_int(q.get("height"))
|
||
frames = _to_int(q.get("frames"))
|
||
fps = _to_float(q.get("fps"))
|
||
|
||
data = recommend_settings(width=width, height=height, frames=frames, fps=fps)
|
||
logging.getLogger("IAMCCS.API").info(
|
||
"[iamccs/hw_probe] cuda=%s vram_gb=%s ram_gb=%s profile=%s vae_tile=%s frames=%s fps=%s",
|
||
data.get("hardware", {}).get("cuda_available"),
|
||
data.get("hardware", {}).get("cuda_total_vram_gb"),
|
||
data.get("hardware", {}).get("system_ram_gb"),
|
||
data.get("recommendations", {}).get("hw_supporter", {}).get("profile"),
|
||
data.get("recommendations", {}).get("vae_decode", {}).get("tile_size"),
|
||
frames,
|
||
fps,
|
||
)
|
||
return web.json_response(data)
|
||
except Exception as e:
|
||
return web.json_response({"error": str(e)}, status=500)
|
||
|
||
except Exception as e:
|
||
# Never hard-fail ComfyUI startup due to optional API endpoints.
|
||
logging.getLogger("IAMCCS.API").warning("Could not setup IAMCCS API routes: %r", e)
|
||
|
||
|
||
# Setup API routes when extension loads
|
||
setup_api_routes()
|
||
|
||
# Print banner after we are fully imported and mappings exist.
|
||
_print_startup_banner()
|
||
|
||
|
||
def _iamccs_install_ltx2_vae_encode_autofix() -> None:
|
||
"""Prevents hard-crash when LTX-2 VAE receives invalid frame counts.
|
||
|
||
Lightricks video VAE encode requires a frame count of the form 1 + 8*x.
|
||
Some workflows can produce off-by-a-few batches (e.g. 240 instead of 241),
|
||
which otherwise raises ValueError and stops execution.
|
||
|
||
This patch pads by repeating the last frame up to the next valid count.
|
||
Opt-in via IAMCCS_LTX2_VAE_ENCODE_AUTOFIX=1.
|
||
"""
|
||
|
||
# Default OFF: user requested workflow-level fixes without monkeypatching VAE.
|
||
if str(os.getenv("IAMCCS_LTX2_VAE_ENCODE_AUTOFIX", "0")).strip().lower() in {"0", "false", "no", "off"}:
|
||
return
|
||
|
||
log = logging.getLogger("IAMCCS.LTX2.VAE")
|
||
|
||
try:
|
||
import torch
|
||
except Exception:
|
||
return
|
||
|
||
try:
|
||
from comfy.ldm.lightricks.vae import causal_video_autoencoder as _cvae
|
||
except Exception:
|
||
# ComfyUI / LTXVideo not installed or import path changed.
|
||
return
|
||
|
||
cls = getattr(_cvae, "CausalVideoAutoencoder", None)
|
||
if cls is None:
|
||
return
|
||
|
||
orig_encode = getattr(cls, "encode", None)
|
||
if orig_encode is None:
|
||
return
|
||
|
||
if getattr(orig_encode, "__iamccs_ltx2_autofix__", False):
|
||
return
|
||
|
||
def _round_up_8n1(frames: int) -> int:
|
||
frames = int(frames)
|
||
if frames <= 1:
|
||
return 1
|
||
rem = (frames - 1) % 8
|
||
if rem == 0:
|
||
return frames
|
||
return frames + (8 - rem)
|
||
|
||
def _is_valid_8n1(frames: int) -> bool:
|
||
frames = int(frames)
|
||
return frames >= 1 and (frames - 1) % 8 == 0
|
||
|
||
def _pad_repeat_last(x: "torch.Tensor", dim: int, pad: int) -> "torch.Tensor":
|
||
# Take last slice along `dim` (keeps dimension) and repeat it `pad` times.
|
||
slc = [slice(None)] * x.ndim
|
||
slc[dim] = slice(-1, None)
|
||
last = x[tuple(slc)]
|
||
reps = [1] * x.ndim
|
||
reps[dim] = int(pad)
|
||
last_rep = last.repeat(*reps)
|
||
return torch.cat([x, last_rep], dim=dim)
|
||
|
||
def _candidate_frame_dims(x: "torch.Tensor") -> list[int]:
|
||
# Most common layouts:
|
||
# - (B, C, T, H, W) -> frames dim = 2
|
||
# - (T, H, W, C) -> frames dim = 0 (ComfyUI IMAGE batches)
|
||
# We only try dims that are >1 and *not obviously channels*.
|
||
dims: list[int] = []
|
||
if x.ndim == 5:
|
||
# Prefer T, then fallbacks
|
||
dims = [2, 0, 1]
|
||
elif x.ndim == 4:
|
||
dims = [0]
|
||
else:
|
||
dims = [0]
|
||
|
||
out: list[int] = []
|
||
for d in dims:
|
||
try:
|
||
size = int(x.shape[d])
|
||
except Exception:
|
||
continue
|
||
if size <= 1:
|
||
continue
|
||
# Heuristic: channels are usually small (1..4). Don't treat that as frames.
|
||
if size in (1, 2, 3, 4) and x.ndim >= 4 and d in (1, 3):
|
||
continue
|
||
out.append(d)
|
||
# Ensure uniqueness, preserve order
|
||
seen = set()
|
||
unique: list[int] = []
|
||
for d in out:
|
||
if d in seen:
|
||
continue
|
||
seen.add(d)
|
||
unique.append(d)
|
||
return unique
|
||
|
||
def encode_patched(self, pixels_in: "torch.Tensor"):
|
||
try:
|
||
return orig_encode(self, pixels_in)
|
||
except ValueError as e:
|
||
msg = str(e)
|
||
if "Invalid number of frames" not in msg:
|
||
raise
|
||
|
||
if not isinstance(pixels_in, torch.Tensor) or pixels_in.ndim < 4:
|
||
raise
|
||
|
||
# Try padding along the most likely frame dimension(s).
|
||
last_err: Exception | None = e
|
||
for dim in _candidate_frame_dims(pixels_in):
|
||
frames_in = int(pixels_in.shape[dim])
|
||
if _is_valid_8n1(frames_in):
|
||
continue
|
||
|
||
frames_fixed = _round_up_8n1(frames_in)
|
||
pad = frames_fixed - frames_in
|
||
if pad <= 0:
|
||
continue
|
||
|
||
try:
|
||
pixels_fixed = _pad_repeat_last(pixels_in, dim=dim, pad=pad)
|
||
log.warning(
|
||
"[LTX2 VAE encode autofix] Padded frames dim=%d %d -> %d (pad=%d) to satisfy 1+8*x rule",
|
||
dim,
|
||
frames_in,
|
||
frames_fixed,
|
||
pad,
|
||
)
|
||
return orig_encode(self, pixels_fixed)
|
||
except Exception as ee:
|
||
last_err = ee
|
||
continue
|
||
|
||
# If all attempts failed, re-raise the original ValueError.
|
||
raise e
|
||
|
||
encode_patched.__iamccs_ltx2_autofix__ = True
|
||
setattr(cls, "encode", encode_patched)
|
||
|
||
|
||
_iamccs_install_ltx2_vae_encode_autofix()
|