ComfyUI_ExtraModels/__init__.py

# 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)

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