Manage GPU memory across LightX2V and ComfyUI models

2026-10-25 22:53:34 +08:00
parent 2df4f7b7ff
commit 9edac027ac
9 changed files with 435 additions and 35 deletions
@@ -44,5 +44,8 @@ _setup_env()
 _register_lightx2v_submodule()

 from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS  # noqa: E402
+from .nodes._runner_memory import install_comfyui_memory_hook  # noqa: E402
+
+install_comfyui_memory_hook()

 __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]