17 lines
795 B
Python
17 lines
795 B
Python
"""ComfyUI_UniBlockSwap.
|
|
|
|
NOTE: the legacy_vram import below MUST stay before the node import - it arms the
|
|
DynamicVRAM bypass, which can only be switched before a model object is built.
|
|
In the default mode ("auto") it does NOT change anything at import time: it only
|
|
installs a /prompt hook, so DynamicVRAM is bypassed solely for workflows that
|
|
actually use this node. Use UNIBLOCKSWAP_LEGACY_VRAM=1 to force the old
|
|
"bypass the whole process at startup" behaviour. See legacy_vram.py.
|
|
"""
|
|
from .legacy_vram import apply as _apply_legacy_vram
|
|
|
|
_apply_legacy_vram() # ComfyUI >= 0.35 + switch ON -> use the legacy path
|
|
|
|
from .uniblockswap_node import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS # noqa: E402
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|