FilesExpand file tree

main

/

__init__.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

51 lines (39 loc) · 1.39 KB

main

/

__init__.py

Copy path

Top

File metadata and controls

• Code
• Blame

51 lines (39 loc) · 1.39 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

# only import if running as a custom node

try:

import comfy.utils

except ImportError:

pass

else:

NODE_CLASS_MAPPINGS = {}

# Deci Diffusion

# from .DeciDiffusion.nodes import NODE_CLASS_MAPPINGS as DeciDiffusion_Nodes

# NODE_CLASS_MAPPINGS.update(DeciDiffusion_Nodes)

# DiT

from .DiT.nodes import NODE_CLASS_MAPPINGS as DiT_Nodes

NODE_CLASS_MAPPINGS.update(DiT_Nodes)

# PixArt

from .PixArt.nodes import NODE_CLASS_MAPPINGS as PixArt_Nodes

NODE_CLASS_MAPPINGS.update(PixArt_Nodes)

# T5

from .T5.nodes import NODE_CLASS_MAPPINGS as T5_Nodes

NODE_CLASS_MAPPINGS.update(T5_Nodes)

# HYDiT

from .HunYuanDiT.nodes import NODE_CLASS_MAPPINGS as HunYuanDiT_Nodes

NODE_CLASS_MAPPINGS.update(HunYuanDiT_Nodes)

# VAE

from .VAE.nodes import NODE_CLASS_MAPPINGS as VAE_Nodes

NODE_CLASS_MAPPINGS.update(VAE_Nodes)

# MiaoBi

from .MiaoBi.nodes import NODE_CLASS_MAPPINGS as MiaoBi_Nodes

NODE_CLASS_MAPPINGS.update(MiaoBi_Nodes)

# Extra

from .utils.nodes import NODE_CLASS_MAPPINGS as Extra_Nodes

NODE_CLASS_MAPPINGS.update(Extra_Nodes)

# Sana

from .Sana.nodes import NODE_CLASS_MAPPINGS as Sana_Nodes

NODE_CLASS_MAPPINGS.update(Sana_Nodes)

# Gemma

from .Gemma.nodes import NODE_CLASS_MAPPINGS as Gemma_Nodes

NODE_CLASS_MAPPINGS.update(Gemma_Nodes)

NODE_DISPLAY_NAME_MAPPINGS = {k:v.TITLE for k,v in NODE_CLASS_MAPPINGS.items()}

__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']