FilesExpand file tree

main

/

__init__.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

75 lines (67 loc) · 2.59 KB

main

/

__init__.py

Copy path

Top

File metadata and controls

• Code
• Blame

75 lines (67 loc) · 2.59 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

try:

from .utils import check_duplicate_nodes, log, color_text

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" - {color_text(dir_path, 'yellow')}\n"

log.warning(color_text(warning_msg + "Please remove duplicates to avoid possible conflicts.", "red"))

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"),

(".LongVie2.nodes", "LongVie2"),

]

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"]