samizdat
d112928e44
Implements wrapper nodes that solve multi-GPU device conflicts for ControlNet preprocessors by temporarily overriding device placement during model loading. Core features: - MultiGPUPreprocessorWrapper base class with device override logic - 5 specific wrapper instances (DepthAnything, DWPose, Canny, OpenPose, Midas) - Graceful ImportError handling with conditional registration - Try/finally blocks ensuring device function restoration - Drop-in replacements maintaining identical INPUT_TYPES/RETURN_TYPES
3 lines
131 B
Python
3 lines
131 B
Python
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |