Meisoftcoltd
1e36f9e915
- Created `MeisoftVRAMDefragmenter` in `vram_node.py` as a passthrough to force model unloading, Python garbage collection, and CUDA cache defragmentation. - Registered the new node in `__init__.py`. - Added light memory cleanup on intermediate batch loops in `loop.py`, alongside a heavy cleanup (including ctypes `malloc_trim`) at the end. - Added `unload_detector_after_analysis` boolean flag to `VideoAnalyzerWithAudio` in `video.py` to allow releasing the BBOX detector model immediately after use.
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
__version__ = "1.5.1"
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
def register_node(c, display_name=None):
|
|
NODE_CLASS_MAPPINGS[c.__name__] = c
|
|
NODE_DISPLAY_NAME_MAPPINGS[c.__name__] = display_name or c.__name__
|
|
return c
|
|
|
|
# Importar tools
|
|
from . import image, loop, video, tools, vram_node
|
|
from .switch_node import MasterSwitch
|
|
|
|
NODE_CLASS_MAPPINGS.update({
|
|
"MasterSwitch": MasterSwitch,
|
|
})
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS.update({
|
|
"MasterSwitch": "🔀 Master Switch",
|
|
"MeisoftVRAMDefragmenter": "🧹 VRAM Defragmenter",
|
|
"SessionImageReceiver": "📥 Session Image Receiver",
|
|
"SessionImageSender": "📤 Session Image Sender",
|
|
"IncrementalVideoStitcher": "🎞️ Incremental Auto-Stitcher",
|
|
"VideoAnalyzerWithAudio": "🕵️ Video Analyzer Face detector + Audio",
|
|
"AutoLoopCalculator": "📊 Auto Loop Calculator",
|
|
"AutoLoopCalculatorWan": "📊 Auto Loop Calculator (WanVideo 3dVAE)",
|
|
"AutoLoopCalculatorLTX": "📊 Auto Loop Calculator (LTX 2.3)",
|
|
"SequentialLoopStart": "🏁 Loop Start (Index)",
|
|
"SequentialLoopTrigger": "🚀 Loop Trigger (Auto-Queue)",
|
|
# Tools
|
|
"AutoFPSLimiter": "⏱️ Auto FPS Limiter",
|
|
"ResTool8x": "📐 ResTool 8x (SD1.5)",
|
|
"ResTool16x": "📏 ResTool 16x (SDXL)",
|
|
"ResTool32x": "🎞️ ResTool 32x (WanVideo)",
|
|
"ResTool64x": "🎬 ResTool 64x (Hunyuan)",
|
|
"ResTool64xLTX": "🌌 ResTool 64x (LTX 2.3)",
|
|
"ConditionalAudioRouter": "🎛️ Conditional Audio Router (Bypass)",
|
|
})
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|