| 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'] |
|
|