kijai
5360eeb345
commitfd32b14fdcAuthor: kijai <40791699+kijai@users.noreply.github.com> Date: Tue Dec 23 02:02:31 2025 +0200 Clean prints commit1776695e26Author: kijai <40791699+kijai@users.noreply.github.com> Date: Tue Dec 23 01:48:12 2025 +0200 Update nodes_model_loading.py commitef36204fa8Author: kijai <40791699+kijai@users.noreply.github.com> Date: Tue Dec 23 01:35:08 2025 +0200 Reduce peak VRAM use commitc6f32c1424Author: kijai <40791699+kijai@users.noreply.github.com> Date: Mon Dec 22 23:53:41 2025 +0200 Norm dtype commit6d4a0f6e53Merge:e7e00063e45021Author: kijai <40791699+kijai@users.noreply.github.com> Date: Mon Dec 22 22:11:38 2025 +0200 Merge branch 'main' into longcat_avatar commite7e00061e5Author: kijai <40791699+kijai@users.noreply.github.com> Date: Mon Dec 22 00:43:01 2025 +0200 Update nodes_sampler.py commiteb5ec262a0Merge:7c0ba84fed3b22Author: kijai <40791699+kijai@users.noreply.github.com> Date: Mon Dec 22 00:42:53 2025 +0200 Merge branch 'main' into longcat_avatar commit7c0ba84a26Author: kijai <40791699+kijai@users.noreply.github.com> Date: Sun Dec 21 23:00:43 2025 +0200 remove prints commit06a86923e7Author: kijai <40791699+kijai@users.noreply.github.com> Date: Sun Dec 21 22:53:25 2025 +0200 Fix ref latent oops commitdca3106f10Author: kijai <40791699+kijai@users.noreply.github.com> Date: Sat Dec 20 18:46:32 2025 +0200 Expose more options, make vid2vid easier commit175418b8d2Author: kijai <40791699+kijai@users.noreply.github.com> Date: Sat Dec 20 03:15:24 2025 +0200 Create LongCatAvatar_testing_wip.json commit4a6e2d3c6cAuthor: kijai <40791699+kijai@users.noreply.github.com> Date: Sat Dec 20 03:14:49 2025 +0200 Init
74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
try:
|
|
from .utils import check_duplicate_nodes, log
|
|
duplicate_dirs = check_duplicate_nodes()
|
|
if duplicate_dirs:
|
|
warning_msg = f"WARNING: Found {len(duplicate_dirs)} other WanVideoWrapper directories:\n"
|
|
for dir_path in duplicate_dirs:
|
|
warning_msg += f" - {dir_path}\n"
|
|
log.warning(warning_msg + "Please remove duplicates to avoid possible conflicts.")
|
|
except:
|
|
pass
|
|
|
|
from .utils import log
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
# Required modules (will raise on import failure)
|
|
REQUIRED_MODULES = [
|
|
(".nodes", "Main"),
|
|
(".nodes_sampler", "Sampler"),
|
|
(".nodes_model_loading", "ModelLoading"),
|
|
(".nodes_utility", "Utility"),
|
|
(".cache_methods.nodes_cache", "Cache"),
|
|
]
|
|
|
|
# Optional modules (will warn on import failure)
|
|
OPTIONAL_MODULES = [
|
|
(".nodes_deprecated", "Deprecated"),
|
|
(".s2v.nodes", "S2V"),
|
|
(".FlashVSR.flashvsr_nodes", "FlashVSR"),
|
|
(".mocha.nodes", "Mocha"),
|
|
(".fun_camera.nodes", "FunCamera"),
|
|
(".uni3c.nodes", "Uni3C"),
|
|
(".controlnet.nodes", "ControlNet"),
|
|
(".ATI.nodes", "ATI"),
|
|
(".multitalk.nodes", "MultiTalk"),
|
|
(".recammaster.nodes", "RecamMaster"),
|
|
(".skyreels.nodes", "SkyReels"),
|
|
(".fantasytalking.nodes", "FantasyTalking"),
|
|
(".qwen.qwen", "Qwen"),
|
|
(".fantasyportrait.nodes", "FantasyPortrait"),
|
|
(".unianimate.nodes", "UniAnimate"),
|
|
(".MTV.nodes", "MTV"),
|
|
(".HuMo.nodes", "HuMo"),
|
|
(".lynx.nodes", "Lynx"),
|
|
(".Ovi.nodes_ovi", "Ovi"),
|
|
(".steadydancer.nodes", "SteadyDancer"),
|
|
(".onetoall.nodes", "OneToAll"),
|
|
(".WanMove.nodes", "WanMove"),
|
|
(".SCAIL.nodes", "SCAIL"),
|
|
(".LongCat.nodes", "LongCat"),
|
|
]
|
|
|
|
def register_nodes(module_path: str, name: str, optional: bool) -> None:
|
|
"""Import and register nodes from a module."""
|
|
try:
|
|
import importlib
|
|
module = importlib.import_module(module_path, package=__package__)
|
|
NODE_CLASS_MAPPINGS.update(getattr(module, "NODE_CLASS_MAPPINGS", {}))
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(getattr(module, "NODE_DISPLAY_NAME_MAPPINGS", {}))
|
|
except Exception as e:
|
|
if optional:
|
|
log.warning(f"WanVideoWrapper WARNING: {name} nodes not available: {e}")
|
|
else:
|
|
raise
|
|
|
|
# Register all node modules
|
|
for module_path, name in REQUIRED_MODULES:
|
|
register_nodes(module_path, name, optional=False)
|
|
|
|
for module_path, name in OPTIONAL_MODULES:
|
|
register_nodes(module_path, name, optional=True)
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] |