| from .nodes import * | |
| NODE_CONFIG = { | |
| "LoadCatvtonFlux": {"class": LoadCatvtonFlux, "name": "Load Catvton Flux"}, | |
| "CatvtonFluxSampler": {"class": CatvtonFluxSampler, "name": "Sample Catvton Flux"}, | |
| "ModelPrinter": {"class": ModelPrinter, "name": "Print Model"}, | |
| "LoadCatvtonFluxLoRA": {"class": LoadCatvtonFluxLoRA, "name": "Load Catvton Flux LoRA"}, | |
| } | |
| def generate_node_mappings(node_config): | |
| node_class_mappings = {} | |
| node_display_name_mappings = {} | |
| for node_name, node_info in node_config.items(): | |
| node_class_mappings[node_name] = node_info["class"] | |
| node_display_name_mappings[node_name] = node_info.get("name", node_info["class"].__name__) | |
| return node_class_mappings, node_display_name_mappings | |
| NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS = generate_node_mappings(NODE_CONFIG) | |
| __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] | |