PixArt initial rewrite

2024-12-10 22:19:02 +01:00
parent 7dc1403cd5
commit 3f844cac42
28 changed files with 583 additions and 3066 deletions
@@ -1,51 +1,50 @@
 # only import if running as a custom node
 try:
-	import comfy.utils
+    import comfy.utils
 except ImportError:
-	pass
+    pass
 else:
-	NODE_CLASS_MAPPINGS = {}
+    NODE_CLASS_MAPPINGS = {}

-	# Deci Diffusion
-	# from .DeciDiffusion.nodes import NODE_CLASS_MAPPINGS as DeciDiffusion_Nodes
-	# NODE_CLASS_MAPPINGS.update(DeciDiffusion_Nodes)
+    # All text encoders
+    from .text_encoders.nodes import NODE_CLASS_MAPPINGS as Tenc_Nodes
+    NODE_CLASS_MAPPINGS.update(Tenc_Nodes)

-	# DiT
-	from .DiT.nodes import NODE_CLASS_MAPPINGS as DiT_Nodes
-	NODE_CLASS_MAPPINGS.update(DiT_Nodes)
+    # Generic nodes
+    from .nodes import NODE_CLASS_MAPPINGS as Base_Nodes
+    NODE_CLASS_MAPPINGS.update(Base_Nodes)

-	# PixArt
-	from .PixArt.nodes import NODE_CLASS_MAPPINGS as PixArt_Nodes
-	NODE_CLASS_MAPPINGS.update(PixArt_Nodes)
+    # DiT
+    from .DiT.nodes import NODE_CLASS_MAPPINGS as DiT_Nodes
+    NODE_CLASS_MAPPINGS.update(DiT_Nodes)

-	# T5
-	from .T5.nodes import NODE_CLASS_MAPPINGS as T5_Nodes
-	NODE_CLASS_MAPPINGS.update(T5_Nodes)
+    # PixArt
+    from .PixArt.nodes import NODE_CLASS_MAPPINGS as PixArt_Nodes
+    NODE_CLASS_MAPPINGS.update(PixArt_Nodes)

-	# HYDiT
-	from .HunYuanDiT.nodes import NODE_CLASS_MAPPINGS as HunYuanDiT_Nodes
-	NODE_CLASS_MAPPINGS.update(HunYuanDiT_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)
+    # 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)
+    # 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)
+    # 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)
+    # 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']
+    # 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']