| | import importlib |
| | import os |
| |
|
| | node_list = [ |
| | "advanced_model_merging", |
| | "reference_only", |
| | "sampler_rescalecfg", |
| | "sampler_tonemap", |
| | "sampler_tonemap_rescalecfg", |
| | "sdxl_model_merging" |
| | ] |
| |
|
| | NODE_CLASS_MAPPINGS = {} |
| | NODE_DISPLAY_NAME_MAPPINGS = {} |
| |
|
| | for module_name in node_list: |
| | imported_module = importlib.import_module(".{}".format(module_name), __name__) |
| |
|
| | NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **imported_module.NODE_CLASS_MAPPINGS} |
| | if hasattr(imported_module, "NODE_DISPLAY_NAME_MAPPINGS"): |
| | NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **imported_module.NODE_DISPLAY_NAME_MAPPINGS} |
| |
|
| | __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |
| |
|