DiT Support

__init__.py

@@ -0,0 +1,18 @@
+# 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)
+	
+	NODE_DISPLAY_NAME_MAPPINGS = {k:v.TITLE for k,v in NODE_CLASS_MAPPINGS.items()}
+	__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']