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